Rajagopal commited on
Commit
e91f526
·
1 Parent(s): 7233a34

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -5
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 of two images and textlist":
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 of two images and textlist",
135
  ],
136
  type="value",
137
- default="image-text",
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