Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -111,14 +111,15 @@ def inference(
|
|
111 |
task,
|
112 |
text_list=None,
|
113 |
image=None,
|
114 |
-
image2=None,
|
115 |
audio=None,
|
|
|
|
|
116 |
):
|
117 |
if task == "image-text":
|
118 |
result = image_text_zeroshot(image, text_list)
|
119 |
elif task == "audio-text":
|
120 |
result = audio_text_zeroshot(audio, text_list)
|
121 |
-
elif task == "embeddings
|
122 |
result = doubleimage_text_zeroshot(image, image2, text_list)
|
123 |
else:
|
124 |
raise NotImplementedError
|
@@ -131,16 +132,16 @@ def main():
|
|
131 |
choices=[
|
132 |
"image-text",
|
133 |
"audio-text",
|
134 |
-
"embeddings
|
135 |
],
|
136 |
type="value",
|
137 |
-
default="
|
138 |
label="Task",
|
139 |
),
|
140 |
gr.inputs.Textbox(lines=1, label="Candidate texts"),
|
141 |
gr.inputs.Image(type="filepath", label="Input image"),
|
142 |
-
gr.inputs.Image(type="filepath", label="Input image2"),
|
143 |
gr.inputs.Audio(type="filepath", label="Input audio"),
|
|
|
144 |
|
145 |
]
|
146 |
|
|
|
111 |
task,
|
112 |
text_list=None,
|
113 |
image=None,
|
|
|
114 |
audio=None,
|
115 |
+
image2=None,
|
116 |
+
|
117 |
):
|
118 |
if task == "image-text":
|
119 |
result = image_text_zeroshot(image, text_list)
|
120 |
elif task == "audio-text":
|
121 |
result = audio_text_zeroshot(audio, text_list)
|
122 |
+
elif task == "embeddings":
|
123 |
result = doubleimage_text_zeroshot(image, image2, text_list)
|
124 |
else:
|
125 |
raise NotImplementedError
|
|
|
132 |
choices=[
|
133 |
"image-text",
|
134 |
"audio-text",
|
135 |
+
"embeddings",
|
136 |
],
|
137 |
type="value",
|
138 |
+
default="embeddings",
|
139 |
label="Task",
|
140 |
),
|
141 |
gr.inputs.Textbox(lines=1, label="Candidate texts"),
|
142 |
gr.inputs.Image(type="filepath", label="Input image"),
|
|
|
143 |
gr.inputs.Audio(type="filepath", label="Input audio"),
|
144 |
+
gr.inputs.Image(type="filepath", label="Input image2"),
|
145 |
|
146 |
]
|
147 |
|