prateekbh commited on
Commit
029ae8b
·
verified ·
1 Parent(s): e0c81f0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -2
app.py CHANGED
@@ -8,8 +8,6 @@ from transformers import StoppingCriteria, TextIteratorStreamer, StoppingCriteri
8
 
9
  device = "cuda:0" if torch.cuda.is_available() else "cpu"
10
 
11
- args = parser.parse_args()
12
- args.low_resource = True
13
 
14
  title = """<h1 style="text-align: center;">Product description generator</h1>"""
15
  css = """
 
8
 
9
  device = "cuda:0" if torch.cuda.is_available() else "cpu"
10
 
 
 
11
 
12
  title = """<h1 style="text-align: center;">Product description generator</h1>"""
13
  css = """