Update app.py

app.py

@@ -1,12 +1,34 @@
-import gradio as gr
+from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
+
+tokenizerModelName = 'google/flan-t5-base'
+instruct_model_name='truocpham/flan-dialogue-summary-checkpoint'
+
+tokenizer = AutoTokenizer.from_pretrained(tokenizerModelName)
+model = AutoModelForSeq2SeqLM.from_pretrained(instruct_model_name)
+
+def SummarizeThis(dialogue):
+  prompt = f"""
+  Summarize the following conversation.
+
+  {dialogue}
 
-def greet(name, intensity):
-    return "Hello, " + name + "!" * int(intensity)
+  Summary:
+  """
 
-demo = gr.Interface(
-    fn=greet,
-    inputs=["text", "slider"],
-    outputs=["text"],
-)
+  inputs = tokenizer(prompt, return_tensors='pt')
+  output = tokenizer.decode(
+      model.generate(
+          inputs["input_ids"], 
+          max_new_tokens=200,
+      )[0], 
+      skip_special_tokens=True
+  )
+
+  return output
+
+
+# Making the gradio application
+import gradio as gr
+iface = gr.Interface(fn=SummarizeThis, inputs="text", outputs=["text"], title="Summarization")
 
-demo.launch()
+iface.launch()