Spaces:

Omnibus
/

chatbots-zero

Running

Omnibus commited on Mar 8

Commit

abdd62e

•

1 Parent(s): 8725491

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,6 +1,7 @@
 import spaces
 from transformers import pipeline
 import gradio as gr
 pipe = pipeline('text-generation', model=model, tokenizer=tokenizer, device='cuda:0')

 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')