Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
|
@@ -72,7 +72,7 @@ else: # download all models
|
|
| 72 |
if torch.cuda.is_available():
|
| 73 |
pipe = pipe.to("cuda")
|
| 74 |
|
| 75 |
-
device = "
|
| 76 |
|
| 77 |
def error_str(error, title="Error"):
|
| 78 |
return f"""#### {title}
|
|
|
|
| 72 |
if torch.cuda.is_available():
|
| 73 |
pipe = pipe.to("cuda")
|
| 74 |
|
| 75 |
+
device = "CPU 🔥" if torch.cuda.is_available() else "CPU 🥶"
|
| 76 |
|
| 77 |
def error_str(error, title="Error"):
|
| 78 |
return f"""#### {title}
|