Update app.py

app.py

@@ -1,32 +1,20 @@
 from transformers import pipeline 
 import torch
+import gradio as gr
 
+summarizer = pipeline(task="summarization",
+                      model="./models/facebook/bart-large-cnn",
+                      torch_dtype=torch.bfloat16)
 
-#translator = pipeline(task="translation",
-#                      model="./models/facebook/nllb-200-distilled-600M",
-#                      torch_dtype=torch.bfloat16) #
-
-#text = """\
-#My puppy is adorable, \
-#Your kitten is cute.
-#Her panda is friendly.
-#His llama is thoughtful. \
-#We all have nice pets!"""
-
-
-#text_translated = translator(text,
-#                             src_lang="eng_Latn",
-#                             tgt_lang="fra_Latn")#
-#
-#text_translated
 
-import gradio as gr
 
 # name of the function
-def greet(name):
-    return "Hello " + name
+def summary(text):
+    return summary = summarizer(text,
+                     min_length=10,
+                     max_length=100)
 
 
-demo = gr.Interface(fn=greet, inputs="text", outputs="text")
+demo = gr.Interface(fn=summary, inputs="text", outputs="text")
 
 demo.launch()