IgorSense commited on
Commit
55627e9
·
1 Parent(s): 2f74ba6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 = "GPU 🔥" if torch.cuda.is_available() else "CPU 🥶"
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}