Update app.py

app.py

@@ -229,7 +229,7 @@ def classify(UserInput, Image, Textbox2, Textbox3):
                 print(f'Maximum Prediction: {max_label} with a value of {max_rounded_prediction}')
     
                 time.sleep(1)
-                if rounded_value > 0.5:
+                if max_rounded_prediction > 0.5:
                     print("\nWays to dispose of this waste: " + max_label)
                     messages.append({"role": "user", "content": content + " " + max_label})
         
@@ -246,9 +246,9 @@ def classify(UserInput, Image, Textbox2, Textbox3):
                     reply = response["choices"][0]["message"]["content"]
                     messages.append({"role": "assistant", "content": reply})
                     
-                    output.append({"Mode": "Image", "type": max_label, "prediction_value": rounded_value, "content": reply})
-                elif rounded_value < 0.5:
-                    output.append({"Mode": "Image", "type": "Not predictable", "prediction_value": rounded_value, "content": "Seems like the prediction rate is too low due to that won't be able to predict the type of material. Try again with a cropped image or different one."})
+                    output.append({"Mode": "Image", "type": max_label, "prediction_value": max_rounded_prediction, "content": reply})
+                elif max_rounded_prediction < 0.5:
+                    output.append({"Mode": "Image", "type": "Not predictable", "prediction_value": max_rounded_prediction, "content": "Seems like the prediction rate is too low due to that won't be able to predict the type of material. Try again with a cropped image or different one."})
             
             return output