Update app.py

app.py

@@ -85,14 +85,14 @@ def gen_fn(model_str, prompt, negative_prompt):
         # Ensure the response is an image or image-like object
         if isinstance(image_response, gr.Image):
             return image_response
-#        elif isinstance(image_response, str):  # If the response is a path or URL, pass it as a string
-#            return gr.Image(image_response)  # You can handle it based on your model's return type
-#        else:
-#            print(f"Unexpected response type: {type(image_response)}")
-#            return None
+        elif isinstance(image_response, str):  # If the response is a path or URL, pass it as a string
+            return gr.Image(image_response)  # You can handle it based on your model's return type
+        else:
+            print(f"Unexpected response type: {type(image_response)}")
+            return None
     except Exception as e:
         print(f"Error occurred: {e}")
-        return e
+        return None
         #None