Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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',
|