Update app.py

app.py

@@ -9,6 +9,14 @@ tokenizer = AutoTokenizer.from_pretrained("kriton/greek-text-summarization")
 model = AutoModelForSeq2SeqLM.from_pretrained("kriton/greek-text-summarization")
 generator = pipeline("summarization", model="kriton/greek-text-summarization")
 
+import gradio as gr
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+
+# Load the model and tokenizer
+tokenizer = AutoTokenizer.from_pretrained("kriton/greek-text-summarization")
+model = AutoModelForSeq2SeqLM.from_pretrained("kriton/greek-text-summarization")
+
+# Define the summary generation function
 def genarate_summary(article):
     inputs = tokenizer(
         'summarize: ' + article, 
@@ -30,19 +38,14 @@ def genarate_summary(article):
 
     return tokenizer.decode(outputs[0], skip_special_tokens=True)
 
-
-
-def generate_text(prompt):
-    response = generator(generate_summary(prompt), max_length=512, num_return_sequences=1)
-    return response[0]["generated_text"]
-
+# Set up Gradio Interface
 iface = gr.Interface(
-    fn=generate_text,
-    inputs="text",
+    fn=genarate_summary, 
+    inputs="text", 
     outputs="text",
-    title="Remote LLM Text Generation",
-    description="Enter a prompt to generate text from the model."
+    title="Greek Text Summarizer",
+    description="Enter an article in Greek, and this tool will generate a summary."
 )
 
-# Launch the Gradio interface
-iface.launch(share=True)
+# Launch the Gradio Interface
+iface.launch()