ydshieh HF staff commited on
Commit
652cd3d
1 Parent(s): f0c57f7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -223,8 +223,9 @@ def main():
223
 
224
  def generate_predictions(image_input, text_input, do_sample, sampling_topp, sampling_temperature):
225
 
 
 
226
  user_image_path = "/tmp/user_input_test_image.jpg"
227
- # This will be of `.jpg` format
228
  image_input.save(user_image_path)
229
  # This might give different results from the original argument `image_input`
230
  image_input = Image.open(user_image_path)
 
223
 
224
  def generate_predictions(image_input, text_input, do_sample, sampling_topp, sampling_temperature):
225
 
226
+ # Save the image and load it again to match the original Kosmos-2 demo.
227
+ # (https://github.com/microsoft/unilm/blob/f4695ed0244a275201fff00bee495f76670fbe70/kosmos-2/demo/gradio_app.py#L345-L346)
228
  user_image_path = "/tmp/user_input_test_image.jpg"
 
229
  image_input.save(user_image_path)
230
  # This might give different results from the original argument `image_input`
231
  image_input = Image.open(user_image_path)