Update main.py

main.py

@@ -24,7 +24,7 @@ model = AutoModelForCausalLM.from_pretrained(
 tokenizer = AutoTokenizer.from_pretrained(model_id, revision=revision)
 
 def generate_answer(image_bytes, question):
-    img = base64.decodebytes(image_bytes)
+    img = base64.b64decode(image_bytes)
     image = Image.open(BytesIO(img))
     enc_image = model.encode_image(image)
     return model.answer_question(enc_image, question, tokenizer)