Update app.py

app.py

@@ -1,3 +1,18 @@
-import gradio as gr
+from transformers import pipeline
+from streamlit import *
+# Load the desired model using Hugging Face's model hub
+model = pipeline(model="philschmid/bart-large-cnn-samsum")
+def generate_text(input_text):
+    # Set the maximum response size to 100 characters
+    output = model(input_text, max_length=100, do_sample=True)
+    # Access the generated response
+    response = output[0]['summarized_text']
+    return response
 
-gr.Interface.load("models/philschmid/bart-large-cnn-samsum", max_length=100)).launch()
+iface = gr.Interface(
+    fn=generate_text,
+    inputs=gr.inputs.Textbox("Input Text"),
+    outputs="text",
+    title="Text Generation App",
+    description="Enter an input text and get a generated response (limited to 100 characters)."
+)