laqey2050 commited on
Commit
10ceaea
·
verified ·
1 Parent(s): 6d544ac

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -6,7 +6,7 @@ from transformers import pipeline
6
 
7
  text_summary = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6", torch_dtype=torch.bfloat16)
8
  #text_summary = pipeline("summarization", model=model_path,
9
- torch_dtype=torch.bfloat16)
10
 
11
  #model_path = ("../Model/models--sshleifer--distilbart-cnn-12-6/snapshots/a4f8f3ea906ed274767e9906dbaede7531d660ff")
12
  #text_summary = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6", torch_dtype=torch.bfloat16)
@@ -25,6 +25,6 @@ gr.close_all()
25
  demo = gr.Interface(fn=summary,
26
  inputs=[gr.Textbox(label="Input text to summarize",lines=6)],
27
  outputs=[gr.Textbox(label="Summarized text",lines=4)],
28
- title="@GenAILearniverse Project 1: Text Summarizer",
29
  description="THIS APPLICATION WILL BE USED TO SUMMARIZE THE TEXT")
30
  demo.launch()
 
6
 
7
  text_summary = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6", torch_dtype=torch.bfloat16)
8
  #text_summary = pipeline("summarization", model=model_path,
9
+ # torch_dtype=torch.bfloat16)
10
 
11
  #model_path = ("../Model/models--sshleifer--distilbart-cnn-12-6/snapshots/a4f8f3ea906ed274767e9906dbaede7531d660ff")
12
  #text_summary = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6", torch_dtype=torch.bfloat16)
 
25
  demo = gr.Interface(fn=summary,
26
  inputs=[gr.Textbox(label="Input text to summarize",lines=6)],
27
  outputs=[gr.Textbox(label="Summarized text",lines=4)],
28
+ title="@LaqeyInc Project 1: Text Summarizer",
29
  description="THIS APPLICATION WILL BE USED TO SUMMARIZE THE TEXT")
30
  demo.launch()