Update app.py

app.py

@@ -185,6 +185,9 @@ with open("labels.txt", "r") as file:
 
 def classify(UserInput, Image, Textbox2, Textbox3):
     if Textbox3 == code:
+        messages = [
+            {"role": "system", "content": system}
+        ]
         if Image is not None:
             output = []
             image_data = np.array(Image)
@@ -208,10 +211,7 @@ def classify(UserInput, Image, Textbox2, Textbox3):
             Textbox2_edited = [x.strip() for x in Textbox2]
             Textbox2_edited = list(Textbox2_edited)
             Textbox2_edited.append(UserInput)
-            messages = [
-                {"role": "system", "content": system},
-                {"role": "user", "content": UserInput},
-            ]
+            messages.append({"role": "user", "content": UserInput})
         
             for i, label in enumerate(labels):
                 prediction_value = float(prediction[0][i])