Update app.py

app.py

@@ -83,9 +83,15 @@ def predict_class(image):
 
 # UI Design for multy class output
 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(i)} is able to", max_length=100)
+    max_value = max(results)
+    for i in range(len(results)):
+      if results[i] == max_value:
+        name = labels.get(i)
+    
+    result_NLP = model_NLP.generate(f"{name} is able to", max_length=100)
     return output, result_NLP