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