Update app.py

app.py

@@ -1,17 +1,7 @@
 import gradio as gr
-   import torch
-   from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+   from diffusers import DiffusionPipeline
 
-   MODEL_NAME = "cloudqi/cqi_text_to_image_pt_v0"
-   tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
-   model = AutoModelForSeq2SeqLM.from_pretrained(MODEL_NAME)
-
-   def generate_images(description):
-       input_ids = tokenizer.encode(description, return_tensors="pt")
-       # Model generates a batch of one image
-       output = model.generate(input_ids)
-       output_image = output[0].numpy().transpose(1,2,0)
-       return output_image.astype("uint8")
+pipeline = DiffusionPipeline.from_pretrained("cloudqi/cqi_text_to_image_pt_v0")
 
    inputs = gr.inputs.Textbox(prompt="Enter Text Description")
    outputs = gr.outputs.Image(label="Generated Image")