style

app.py

@@ -1,16 +1,21 @@
 import gradio as gr
 from transformers import pipeline
 
-# Load a simple model (fill-mask task as an example)
-model = pipeline("fill-mask", model="distilroberta-base")
+# Load a large model (e.g., text generation)
+model = pipeline("text-generation", model="gpt-3")
 
-# Define the function to use the GPU
-def predict(text):
-    return model(text)
+def generate_text(prompt):
+    return model(prompt, max_length=50)
 
-# Gradio Interface
-interface = gr.Interface(fn=predict, inputs="text", outputs="text")
+# Custom interface with sliders and descriptions
+interface = gr.Interface(
+    fn=generate_text,
+    inputs=gr.Textbox(label="Enter your prompt here"),
+    outputs=gr.Textbox(label="Generated Text"),
+    title="AI Text Generator",
+    description="This app generates text based on your input prompt. Try it out!",
+    theme="dark"
+)
 
-# Launch the interface
 if __name__ == "__main__":
     interface.launch()