Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -56,7 +56,7 @@ def respond(
|
|
56 |
)
|
57 |
|
58 |
if torch.cuda.is_available():
|
59 |
-
inputs = inputs.to(
|
60 |
model.to("cuda")
|
61 |
|
62 |
# Set up the streamer
|
|
|
56 |
)
|
57 |
|
58 |
if torch.cuda.is_available():
|
59 |
+
inputs = inputs.to("cuda")
|
60 |
model.to("cuda")
|
61 |
|
62 |
# Set up the streamer
|