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