Update app.py
Browse files
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 |
|