Upload 3 files

app.py

@@ -0,0 +1,23 @@
+from transformers import pipeline
+import gradio as gr
+
+# Initialize the text generation pipeline
+generator = pipeline("text-generation", model="gpt2")
+
+def generate_response(prompt):
+    """Generate text based on the user's prompt."""
+    response = generator(prompt, max_length=100, num_return_sequences=1)
+    return response[0]["generated_text"]
+
+# Create a Gradio interface
+interface = gr.Interface(
+    fn=generate_response, 
+    inputs="text", 
+    outputs="text",
+    title="Simple LLM",
+    description="Enter a prompt to get a response generated by GPT-2!"
+)
+
+# Launch the app
+if __name__ == "__main__":
+    interface.launch()

requirements.txt

@@ -0,0 +1,3 @@
+transformers
+torch
+gradio