Update app.py

app.py

@@ -114,8 +114,8 @@ class Predictor:
     def predict(
         self,
         image,
-        query = self.class_embed,
-        tag_names = self.tag_names,
+        query,
+        tag_names,
     ):
         image = self.prepare_image(image)
 
@@ -130,12 +130,19 @@ class Predictor:
 
         return general_tag_preds_dict
 
+    def predict_seen_tags(
+        self,
+        image,
+    ):
+        return self.predict(image, self.class_embed, self.tag_names)
+        
+    
     def predict_new_tag(
         self,
         image,
         query,
     ):
-        return self.predict(image, query=query, tag_names=["embedding"])["embedding"]
+        return self.predict(image, query, ["embedding"])["embedding"]
 
 
 def main():
@@ -189,7 +196,7 @@ def main():
                     )
 
         submit.click(
-            predictor.predict,
+            predictor.predict_seen_tags,
             inputs=[
                 image,
             ],