Update app.py

app.py

@@ -85,7 +85,7 @@ def predict_class(image):
 def classify_image(image):
     results = predict_class(image)
     output = {labels.get(i): float(results[i]) for i in range(len(results))}
-    result_NLP = model_NLP.generate(f"{labels.get(0)} is able to", max_length=100)
+    result_NLP = model_NLP.generate(f"{labels.get(i)} is able to", max_length=100)
     return output, result_NLP