Kilos1 commited on
Commit
057300b
·
verified ·
1 Parent(s): 2c4f69d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -71,7 +71,7 @@ def generate_model_response(image_file, user_query):
71
 
72
  input_text = assistant_prompt + "\n\n" + user_query + "\n![Image](data:image/jpeg;base64," + encoded_image + ")"
73
 
74
- inputs = tokenizer(input_text, return_tensors="pt")
75
 
76
  try:
77
  outputs = model.generate(**inputs)
 
71
 
72
  input_text = assistant_prompt + "\n\n" + user_query + "\n![Image](data:image/jpeg;base64," + encoded_image + ")"
73
 
74
+ inputs = AutoTokenizer(input_text, return_tensors="pt")
75
 
76
  try:
77
  outputs = model.generate(**inputs)