mayankchugh-learning commited on
Commit
7f4a5e2
1 Parent(s): c8085f8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -3,11 +3,11 @@ from transformers import pipeline
3
  import torch
4
  import gradio as gr
5
 
6
- # pipe = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6", torch_dtype=torch.bfloat16)
7
 
8
- model_path = "Model/models--sshleifer--distilbart-cnn-12-6/snapshots/a4f8f3ea906ed274767e9906dbaede7531d660ff"
9
 
10
- text_summary = pipeline("summarization", model=model_path, torch_dtype=torch.bfloat16)
11
 
12
  # text="Why You Can Trust Forbes Advisor Small Business \
13
  # The Forbes Advisor Small Business team is committed to bringing you unbiased \
 
3
  import torch
4
  import gradio as gr
5
 
6
+ text_summary = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6", torch_dtype=torch.bfloat16)
7
 
8
+ # model_path = "Model/models--sshleifer--distilbart-cnn-12-6/snapshots/a4f8f3ea906ed274767e9906dbaede7531d660ff"
9
 
10
+ # text_summary = pipeline("summarization", model=model_path, torch_dtype=torch.bfloat16)
11
 
12
  # text="Why You Can Trust Forbes Advisor Small Business \
13
  # The Forbes Advisor Small Business team is committed to bringing you unbiased \