Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
|
@@ -177,6 +177,7 @@ def reset_inference_state():
|
|
| 177 |
def infer(ref_style_file, style_description, caption):
|
| 178 |
global models_rbm, models_b
|
| 179 |
try:
|
|
|
|
| 180 |
height=1024
|
| 181 |
width=1024
|
| 182 |
batch_size=1
|
|
|
|
| 177 |
def infer(ref_style_file, style_description, caption):
|
| 178 |
global models_rbm, models_b
|
| 179 |
try:
|
| 180 |
+
caption = f"{caption} in {style_description}"
|
| 181 |
height=1024
|
| 182 |
width=1024
|
| 183 |
batch_size=1
|