Spaces:
Runtime error
Runtime error
fix prompts
Browse files
app.py
CHANGED
|
@@ -239,6 +239,10 @@ def inference(img, text, is_translation, prompt=None):
|
|
| 239 |
model = model.eval()
|
| 240 |
device = CUDA(0) if is_gpu else "cpu"
|
| 241 |
model = model.to(device)
|
|
|
|
|
|
|
|
|
|
|
|
|
| 242 |
if is_translation:
|
| 243 |
# encode text
|
| 244 |
if text is None:
|
|
@@ -277,7 +281,7 @@ inputs = [
|
|
| 277 |
gr.inputs.Image(type="file", label="Image to generate Vietnamese caption", optional=True),
|
| 278 |
gr.inputs.Textbox(lines=2, placeholder="English sentence for translation"),
|
| 279 |
gr.inputs.Checkbox(),
|
| 280 |
-
gr.inputs.Textbox(lines=1, placeholder="Prompt [Optional]"
|
| 281 |
]
|
| 282 |
|
| 283 |
gr.Interface(
|
|
|
|
| 239 |
model = model.eval()
|
| 240 |
device = CUDA(0) if is_gpu else "cpu"
|
| 241 |
model = model.to(device)
|
| 242 |
+
|
| 243 |
+
promt = prompt.strip()
|
| 244 |
+
if not prompt:
|
| 245 |
+
prompt = None
|
| 246 |
if is_translation:
|
| 247 |
# encode text
|
| 248 |
if text is None:
|
|
|
|
| 281 |
gr.inputs.Image(type="file", label="Image to generate Vietnamese caption", optional=True),
|
| 282 |
gr.inputs.Textbox(lines=2, placeholder="English sentence for translation"),
|
| 283 |
gr.inputs.Checkbox(),
|
| 284 |
+
gr.inputs.Textbox(lines=1, placeholder="Prompt [Optional]")
|
| 285 |
]
|
| 286 |
|
| 287 |
gr.Interface(
|