Update app.py

app.py

@@ -1,3 +1,21 @@
 import gradio as gr
+from transformers import pipeline
 
-gr.load("models/TinyLlama/TinyLlama-1.1B-Chat-v1.0").launch()
+# Load the model using Hugging Face's transformers library
+model_name = "facebook/blenderbot-400M-distill"  # Example model, replace with your desired model
+nlp = pipeline("text-generation", model=model_name)
+
+# Define the function to generate text
+def generate_text(prompt):
+    response = nlp(prompt, max_length=50, num_return_sequences=1)
+    return response[0]['generated_text']
+
+# Create a Gradio interface
+iface = gr.Interface(fn=generate_text, 
+                     inputs="text", 
+                     outputs="text",
+                     title="Hugging Face Model",
+                     description="Generate text using a Hugging Face model.")
+
+# Launch the interface
+iface.launch()