Update app.py

app.py

@@ -207,9 +207,12 @@ def process_image(image_input, user_prompt):
             image_input = image_file.read()
             SaveNewFile=False # file is there and this is just prompt inference
     else:
-        image_file_name = image_input.name
-        image_input = image_input.read()
-        SaveNewFile=True
+        if image_input is None:
+            data=False
+        else:
+            image_file_name = image_input.name
+            image_input = image_input.read()
+            SaveNewFile=True
     
     st.markdown('Processing image: ' + image_file_name)
     base64_image = base64.b64encode(image_input).decode("utf-8")