Update app.py

app.py

@@ -1,20 +1,2 @@
-import gradio as gr
-from transformers import pipeline
-
-# Initialize your model here
-generator = pipeline("image-generation", model="CompVis/stable-diffusion-v1-4")
-
-# Function to generate image from text prompt
-def generate_image(prompt):
-    return generator(prompt)[0]["image"]
-
-# Define Gradio Interface
-interface = gr.Interface(
-    fn=generate_image,
-    inputs="text",  # User will input text prompt
-    outputs="image",  # The output will be an image
-    live=True
-)
-
-# Launch the interface
-interface.launch()
+!pip install transformers
+transformers==4.x.x