Yermek68 commited on
Commit
ad5f036
·
verified ·
1 Parent(s): 8c445f1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +22 -0
app.py CHANGED
@@ -46,3 +46,25 @@ def ping():
46
  if __name__ == "__main__":
47
  import uvicorn
48
  uvicorn.run("app:app", host="0.0.0.0", port=7860, reload=True)
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
46
  if __name__ == "__main__":
47
  import uvicorn
48
  uvicorn.run("app:app", host="0.0.0.0", port=7860, reload=True)
49
+
50
+ import gradio as gr
51
+
52
+ def summarize_text_interface(text):
53
+ from transformers import pipeline
54
+ summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
55
+ summary = summarizer(text, max_length=150, min_length=40, do_sample=False)
56
+ return summary[0]['summary_text']
57
+
58
+ demo = gr.Interface(
59
+ fn=summarize_text_interface,
60
+ inputs=gr.Textbox(lines=10, label="Введите текст для суммаризации"),
61
+ outputs=gr.Textbox(label="Резюме"),
62
+ title="Eroha Agent Summarizer",
63
+ description="Введите текст для краткого резюме с помощью модели."
64
+ )
65
+
66
+ if __name__ == "__main__":
67
+ import threading
68
+ import uvicorn
69
+ threading.Thread(target=lambda: demo.launch(server_name="0.0.0.0", server_port=7860)).start()
70
+ uvicorn.run(app, host="0.0.0.0", port=8000)