Update app.py

app.py

@@ -40,8 +40,7 @@ class FoodQualityAI:
         
         class_names = list(set([os.path.splitext(f)[0].split('_')[0] for f in file_names]))
             
-        predicted_idx = np.argmax(prediction)
-        predicted_class = class_names[predicted_idx]
+        predicted_class = class_names[np.argmax(prediction)]
         confidence = float(np.max(prediction))
 
         return predicted_class, confidence