Update app.py

app.py

@@ -1,6 +1,10 @@
 import os
 import gradio as gr
 from huggingface_hub import login
+from zerogpu import ZeroGPU  # Import ZeroGPU
+
+# Initialize ZeroGPU
+zero_gpu = ZeroGPU()
 
 # Gradio demo definition
 def create_demo():
@@ -18,10 +22,12 @@ def create_demo():
             
             with gr.Column():
                 result = gr.Image(label="Generated Image")
-                
-        # Since this is a zero GPU demo, we'll show a message about the model's unavailability
+        
+        def mock_generation(prompt, negative_prompt):
+            return None  # Return None as we can't generate images in zero GPU environment
+        
         generate.click(
-            fn=lambda x, y: gr.Info("This is a zero GPU demo. The model cannot generate images without GPU resources.").value,
+            fn=mock_generation,
             inputs=[prompt, negative_prompt],
             outputs=result
         )
@@ -32,4 +38,10 @@ def create_demo():
 # Create and launch the demo
 if __name__ == "__main__":
     demo = create_demo()
-    demo.launch()
+    # Configure the launch settings for Spaces
+    demo.launch(
+        server_name="0.0.0.0",  # Required for Spaces
+        server_port=7860,       # Default Spaces port
+        share=False,            # Don't create a public link
+        enable_queue=True       # Enable queuing for better handling of multiple users
+    )