Spaces:

Omnibus
/

chatbots-zero

Running

Omnibus commited on Mar 8

Commit

3649895

•

1 Parent(s): abdd62e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,6 +2,7 @@ import spaces
 from transformers import pipeline
 import gradio as gr
 model = 'mosaicml/mpt-7b-instruct'
 pipe = pipeline('text-generation', model=model, tokenizer=tokenizer, device='cuda:0')

 from transformers import pipeline
 import gradio as gr
 model = 'mosaicml/mpt-7b-instruct'
+tokenizer = 'mosaicml/mpt-7b-instruct'
 pipe = pipeline('text-generation', model=model, tokenizer=tokenizer, device='cuda:0')