andrew3279 commited on
Commit
086b4de
1 Parent(s): 63a9cb3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ print(device)
10
 
11
  model_name = "bigscience/bloom-1b1"
12
 
13
- model = BloomForCausalLM.from_pretrained(model_name,)
14
  tokenizer = BloomTokenizerFast.from_pretrained(model_name)
15
 
16
  # Define the pipeline for Gradio purpose
 
10
 
11
  model_name = "bigscience/bloom-1b1"
12
 
13
+ model = BloomForCausalLM.from_pretrained(model_name)
14
  tokenizer = BloomTokenizerFast.from_pretrained(model_name)
15
 
16
  # Define the pipeline for Gradio purpose