tonyassi commited on
Commit
6a89211
·
verified ·
1 Parent(s): 84bc8e5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -68,6 +68,7 @@ def greet(description,color,features,occasion,type_,face):
68
  prompt += description + color + features + occasion + type_
69
 
70
  print('prompt:',prompt)
 
71
  image = pipe(
72
  prompt,
73
  negative_prompt='deformed face,bad anatomy',
 
68
  prompt += description + color + features + occasion + type_
69
 
70
  print('prompt:',prompt)
71
+ pipe.to("cuda")
72
  image = pipe(
73
  prompt,
74
  negative_prompt='deformed face,bad anatomy',