Update app.py
Browse files
app.py
CHANGED
|
@@ -8,7 +8,7 @@ pipe = pipeline(
|
|
| 8 |
"text-generation",
|
| 9 |
model="sarvamai/sarvam-translate",
|
| 10 |
torch_dtype=torch.bfloat16,
|
| 11 |
-
device="cuda",
|
| 12 |
)
|
| 13 |
|
| 14 |
@spaces.GPU
|
|
|
|
| 8 |
"text-generation",
|
| 9 |
model="sarvamai/sarvam-translate",
|
| 10 |
torch_dtype=torch.bfloat16,
|
| 11 |
+
device="cuda:0",
|
| 12 |
)
|
| 13 |
|
| 14 |
@spaces.GPU
|