Update app.py

app.py

@@ -1,18 +1,21 @@
-from huggingface_hub import snapshot_download
 import gradio as gr
+from transformers import pipeline
 
-# Download the model to cache or local dir
-local_dir = snapshot_download(repo_id="cloudqi/cqi_text_to_image_pt_v0")
+# Load the model with remote code
+pipe = pipeline("text-to-image", model="cloudqi/cqi_text_to_image_pt_v0", trust_remote_code=True)
 
-def show_path():
-    return f"Model downloaded to: {local_dir}"
+def generate_image(prompt):
+    result = pipe(prompt)[0]  # Get the first generated image
+    return result
 
-gr.Interface(fn=show_path, inputs=None, outputs="text").launch()
-demo.launch(share=True)
-
-from diffusers import DiffusionPipeline
+# Create the Gradio interface
+demo = gr.Interface(
+    fn=generate_image,
+    inputs=gr.Textbox(lines=2, placeholder="Enter your prompt here..."),
+    outputs="image",
+    title="Text-to-Image Generator",
+    description="Powered by cloudqi/cqi_text_to_image_pt_v0"
+)
 
-pipe = DiffusionPipeline.from_pretrained("cloudqi/cqi_text_to_image_pt_v0")
-
-prompt = "Gato em alta qualidade na neve\n"
-image = pipe(prompt).images[0]
+# Run with public link enabled
+demo.launch(share=True)