Manel commited on
Commit
ca164c1
·
verified ·
1 Parent(s): 1893b91

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -236,7 +236,7 @@ if __name__=="__main__":
236
 
237
 
238
  # model to use in spaces depends on the available device
239
- device = "cuda" if torch.cuda.is_available() else "cpu
240
 
241
  model_name = "llama" if device=="cpu" else "mistral"
242
 
 
236
 
237
 
238
  # model to use in spaces depends on the available device
239
+ device = "cuda" if torch.cuda.is_available() else "cpu"
240
 
241
  model_name = "llama" if device=="cpu" else "mistral"
242