Update app.py

app.py

@@ -1,11 +1,26 @@
 import gradio as gr
+from transformers import T5Tokenizer, T5ForConditionalGeneration
+
+tokenizer = T5Tokenizer.from_pretrained("./model_files")
+model = T5ForConditionalGeneration.from_pretrained("./model_files")
+
+
 
 
 def greet(name):
-    return "Hello " + name
+    inp =  "Summarize:" + name
+    input_ids = tokenizer(inp, return_tensors="pt").input_ids
+    outputs = model.generate(input_ids,
+                        max_length=150, 
+              num_beams=2,
+              repetition_penalty=2.5, 
+              length_penalty=1.0, 
+              early_stopping=True)
+    return tokenizer.decode(outputs[0],skip_special_tokens=True, clean_up_tokenization_spaces=True))
+    
 
 
 # We instantiate the Textbox class
-textbox = gr.Textbox(label="Type your name here:", placeholder="John Doe", lines=2)
+textbox = gr.Textbox(label="Type your name here:", placeholder="Article", lines=5)
 
 gr.Interface(fn=greet, inputs=textbox, outputs="text").launch()