Spaces:
Running
on
Zero
Running
on
Zero
AbstractPhil
commited on
Commit
·
40292d5
1
Parent(s):
644faa4
yes
Browse files
app.py
CHANGED
@@ -251,6 +251,7 @@ else:
|
|
251 |
try:
|
252 |
if seed is not None:
|
253 |
torch.manual_seed(int(seed))
|
|
|
254 |
|
255 |
# Load model
|
256 |
model = _load_model_on("auto" if torch.cuda.is_available() else "cpu")
|
|
|
251 |
try:
|
252 |
if seed is not None:
|
253 |
torch.manual_seed(int(seed))
|
254 |
+
|
255 |
|
256 |
# Load model
|
257 |
model = _load_model_on("auto" if torch.cuda.is_available() else "cpu")
|