Update app.py

app.py

@@ -72,13 +72,13 @@ class ZeroShotCLF():
 
 clf = ZeroShotCLF()
 
-def clasifica(imagen, clases_, tipos_):
-    clases = clases_.split(", ")
-    tipos = tipos_.split(", ")
+def clasifica(Image, Types, Classes):
+    clases = Classes.split(", ")
+    tipos = Types.split(", ")
     if len(tipos) == 1:
         tipos = tipos[0]
     textos = clf.set_classes(clases, tipos)
     probs = clf.predict(PIL.Image.fromarray(imagen))[-1]
     return {textos[i]: probs[i] for i in range(len(textos))}
 
-gr.Interface(fn=clasifica, inputs=["image", "text", "text"], outputs="text").launch(share=False)
+gr.Interface(fn=clasifica, inputs=["image", "text", "text"], outputs=gr.components.Label()).launch(share=False)