Update app.py

app.py

@@ -86,11 +86,10 @@ def gen_fn(model_str, prompt, negative_prompt):
             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 None
+    except Exception as e:
+        print(f"Error occurred: {e}")
+        return None