Spaces:

madansa7
/

deepcoder-demo

Runtime error

madansa7 commited on 6 days ago

Commit

48eb102

verified ·

1 Parent(s): ad14191

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,7 +4,7 @@ from transformers import pipeline
 pipe = pipeline(
     "text-generation",
     model="agentica-org/DeepCoder-14B-Preview",
-    device="cuda"
 )
 def chat(message, history):

 pipe = pipeline(
     "text-generation",
     model="agentica-org/DeepCoder-14B-Preview",
+    device="cuda" if torch.cuda.is_available() else "cpu"
 )
 def chat(message, history):