Spaces:
Runtime error
Runtime error
Upload multit2i.py
Browse files- multit2i.py +2 -2
multit2i.py
CHANGED
|
@@ -163,12 +163,12 @@ def infer(prompt: str, model_name: str, recom_prompt: bool, progress=gr.Progress
|
|
| 163 |
caption = model_name.split("/")[-1]
|
| 164 |
try:
|
| 165 |
model = load_model(model_name)
|
| 166 |
-
if not model: return (
|
| 167 |
image_path = model(prompt + rprompt + seed)
|
| 168 |
image = Image.open(image_path).convert('RGB')
|
| 169 |
except Exception as e:
|
| 170 |
print(e)
|
| 171 |
-
return (
|
| 172 |
return (image, caption)
|
| 173 |
|
| 174 |
|
|
|
|
| 163 |
caption = model_name.split("/")[-1]
|
| 164 |
try:
|
| 165 |
model = load_model(model_name)
|
| 166 |
+
if not model: return ("", None)
|
| 167 |
image_path = model(prompt + rprompt + seed)
|
| 168 |
image = Image.open(image_path).convert('RGB')
|
| 169 |
except Exception as e:
|
| 170 |
print(e)
|
| 171 |
+
return ("", None)
|
| 172 |
return (image, caption)
|
| 173 |
|
| 174 |
|