Spaces:

LeroyDyer
/

iChat

Runtime error

LeroyDyer commited on Mar 28

Commit

ea3c486

•

1 Parent(s): ec02464

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -5,7 +5,7 @@ import os
 import spaces
 @spaces.GPU
 def load_model(model_name):
-    return pipeline("text-generation", model=model_name, device_map="auto",  trust_remote_code=True, token=os.environ["token"])
 @spaces.GPU()
 def generate(
     model_name,

 import spaces
 @spaces.GPU
 def load_model(model_name):
+    return pipeline("text-generation", model=model_name, device_map="auto",  trust_remote_code=True)
 @spaces.GPU()
 def generate(
     model_name,