Update app.py

app.py

@@ -795,8 +795,9 @@ def classify(platform, UserInput, Images, Textbox2, Textbox3):
                 # base64_image_str = encode_image(image)
 
                 if image_data:
-                    with io.BytesIO(image_data) as buffer:
-                        image = Image.open(buffer)
+                    try:
+                        # Open the image directly from the image data
+                        image = Image.open(io.BytesIO(image_data))
                         base64_image_str = encode_image(image)
                         
                         payload = {
@@ -827,6 +828,9 @@ def classify(platform, UserInput, Images, Textbox2, Textbox3):
                                 pass
                         else:
                             return "Answer: not found in the string."
+
+                    except:
+                        return "ERRRRRRR"
                 else:
                     print("Error: Image data is not available.")
                     return None