Update app.py

app.py

@@ -13,7 +13,7 @@ model.eval()
 def get_tags(image, threshold, top_k):
     current_datetime = datetime.now().strftime("%Y-%m-%d_%H-%M-%S")
     print(f"{current_datetime}: started.")
-    results = tagger.predict(src, topk=topk, threshold=threshold)
+    results = tagger.predict(image, topk=top_k, threshold=threshold)
     temp = []
     return_dict = dict()
     for rank, (tag, score) in enumerate(results, 1):
@@ -27,7 +27,7 @@ def get_tags(image, threshold, top_k):
 demo = gr.Interface(
     get_tags,
     inputs=[
-        gr.Image(label="Source", sources=['upload',], type='pil'), 
+        gr.Image(label="Source", sources=['upload',], type='filepath'), 
         gr.Slider(minimum=0.00, maximum=1.00, step=0.01, value=0.30, label="Threshold"),
         gr.Slider(minimum=0, maximum=500, step=1, value=30, label="Top K")
     ],