Create app.py

app.py

@@ -0,0 +1,20 @@
+import gradio as gr
+from transformers import pipeline
+
+# Load the model
+model = pipeline("text-generation", model="rish13/polymers_llm")
+
+def generate_text(prompt):
+    return model(prompt, max_length=50, num_return_sequences=1)[0]['generated_text']
+
+# Define the Gradio interface
+interface = gr.Interface(
+    fn=generate_text,
+    inputs=gr.Textbox(lines=2, placeholder="Enter your text here..."),
+    outputs="text",
+    title="Text Generation Model",
+    description="A text generation model using Hugging Face Transformers."
+)
+
+# Launch the interface
+interface.launch()