sounar commited on
Commit
a7e3fe6
·
verified ·
1 Parent(s): 451cfa7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -38,7 +38,6 @@ if eos_token_id is None:
38
  raise ValueError(
39
  "Neither `eos_token_id` nor `pad_token_id` is defined in the tokenizer. Please specify one explicitly."
40
  )
41
- model.generation_config.eos_token_id = eos_token_id
42
 
43
  # Preprocess image
44
  def preprocess_image(image):
@@ -70,9 +69,9 @@ def analyze_input(image, question):
70
 
71
  # Generate response
72
  outputs = model.generate(
73
- input_ids=model_inputs["input_ids"],
74
  max_new_tokens=256,
75
- eos_token_id=model.generation_config.eos_token_id
76
  )
77
 
78
  # Decode the response
 
38
  raise ValueError(
39
  "Neither `eos_token_id` nor `pad_token_id` is defined in the tokenizer. Please specify one explicitly."
40
  )
 
41
 
42
  # Preprocess image
43
  def preprocess_image(image):
 
69
 
70
  # Generate response
71
  outputs = model.generate(
72
+ model_inputs=model_inputs, # Explicitly pass the model_inputs
73
  max_new_tokens=256,
74
+ eos_token_id=eos_token_id
75
  )
76
 
77
  # Decode the response