Update app.py

app.py

@@ -1,35 +1,17 @@
 import gradio as gr
-from PIL import Image, ImageDraw, ImageFont
+from transformers import pipeline
 
-def generate_text(prompt, quality="Medium"):
-    # Your text generation code here, using the prompt and quality level inputs
-    # For this example, we'll just return the input prompt and quality level
-    return f"Prompt: {prompt}, Quality: {quality}"
+text_generator = pipeline("text-generation", model="EleutherAI/gpt-neo-2.7B")
 
-def generate_image(text):
-    # Create a new image with a white background
-    img = Image.new('RGB', (500, 500), color='white')
+def generate_text(prompt, quality):
+    length_dict = {"Low": 25, "Medium (Default)": 50, "High": 100}
+    length = length_dict[quality]
+    return text_generator(prompt, max_length=length, do_sample=True)[0]['generated_text']
 
-    # Draw the text onto the image
-    draw = ImageDraw.Draw(img)
-    font = ImageFont.truetype("arial.ttf", size=20)
-    draw.text((10, 10), text, fill='black', font=font)
-
-    # Return the image as a numpy array
-    return img
-
-prompt_input = gr.inputs.Textbox(label="Enter your prompt:")
-quality_dropdown = gr.inputs.Dropdown(choices=["Low", "Medium", "High"], label="Select quality level:", default="Medium")
-output_text = gr.outputs.Textbox(label="Generated text:")
-output_image = gr.outputs.Image(type="pil", label="Generated image:")
-
-iface = gr.Interface(fn=generate_text, inputs=[prompt_input, quality_dropdown], outputs=[output_text, output_image], title="Text and Image Generator")
-
-def update_image(output_text):
-    # Generate a new image based on the generated text
-    img = generate_image(output_text)
-    return img
-
-iface.out[1].update(update_image, "output_text")
+iface = gr.Interface(
+  fn=generate_text, 
+  inputs=["textbox", gr.inputs.Dropdown(["Low", "Medium (Default)", "High"], label="Quality")],
+  outputs=gr.outputs.Image(label="Generated image:")
+)
 
 iface.launch()