Update app.py

app.py

@@ -419,14 +419,12 @@ def classify(platform, UserInput, Images, Textbox2, Textbox3):
                             "model": model_llm
                         }).json()
                         print("RESPONSE TRY",response)
+                        reply = response["choices"][0]["message"]["content"]
+                        messages.append({"role": "assistant", "content": reply})
+                        output.append({"Mode": "Image", "type": max_label, "prediction_value": max_rounded_prediction, "content": reply})
                     except:
                         print("DOESN'T WORK")
-                    
-
-                    reply = response["choices"][0]["message"]["content"]
-                    messages.append({"role": "assistant", "content": reply})
-
-                    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"})