Update app.py

app.py

@@ -1,3 +1,35 @@
+import os
 import gradio as gr
+from huggingface_hub import login
 
-gr.load("models/ostris/Flex.1-alpha"), headers={"X-IP-Token": x_ip_token}.launch()
+# Gradio demo definition
+def create_demo():
+    with gr.Blocks() as demo:
+        gr.Markdown("# Flex.1-alpha Text to Image Generation")
+        
+        with gr.Row():
+            with gr.Column():
+                prompt = gr.Textbox(label="Prompt", placeholder="Enter your prompt here...")
+                negative_prompt = gr.Textbox(label="Negative Prompt", placeholder="What you don't want to see in the image...")
+                
+                with gr.Row():
+                    generate = gr.Button("Generate")
+                    clear = gr.Button("Clear")
+            
+            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
+        generate.click(
+            fn=lambda x, y: gr.Info("This is a zero GPU demo. The model cannot generate images without GPU resources.").value,
+            inputs=[prompt, negative_prompt],
+            outputs=result
+        )
+        clear.click(lambda: ["", ""], outputs=[prompt, negative_prompt])
+        
+    return demo
+
+# Create and launch the demo
+if __name__ == "__main__":
+    demo = create_demo()
+    demo.launch()