app.py

app.py

@@ -0,0 +1,27 @@
+from transformers import BertTokenizer
+import os
+import io
+from IPython.display import Image, display, HTML
+from PIL import Image
+import base64
+import torch
+from transformers import pipeline
+from transformers import AutoModel
+
+acc_token='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'
+
+get_completion = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6")
+def summarize(input):
+    output = get_completion(input)
+    return output[0]['summary_text']
+
+import gradio as gr
+def summarize(input):
+    output = get_completion(input)
+    return output[0]['summary_text']
+    
+gr.close_all()
+
+
+demo = gr.Interface(fn=summarize, inputs="text", outputs="text")
+demo.launch(share=True)