AdrienB134 commited on
Commit
c34d360
1 Parent(s): 28ef079
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -50,6 +50,8 @@ def model_inference(
50
  images, text, assistant_prefix= None, decoding_strategy = "Greedy", temperature= 0.4, max_new_tokens=512,
51
  repetition_penalty=1.2, top_p=0.8
52
  ):
 
 
53
  print(type(images))
54
  if text == "" and not images:
55
  gr.Error("Please input a query and optionally image(s).")
 
50
  images, text, assistant_prefix= None, decoding_strategy = "Greedy", temperature= 0.4, max_new_tokens=512,
51
  repetition_penalty=1.2, top_p=0.8
52
  ):
53
+ print(type(images))
54
+ images = images[0]
55
  print(type(images))
56
  if text == "" and not images:
57
  gr.Error("Please input a query and optionally image(s).")