Update app.py

app.py

@@ -78,7 +78,7 @@ def clasifica(imagen, clases_, tipos_):
     if len(tipos) == 1:
         tipos = tipos[0]
     textos = clf.set_classes(clases, tipos)
-    probs = clf.predict(PIL.Image.fromarray(imagen))[-1][0]
+    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)