refactor: pil

app.py

@@ -25,7 +25,7 @@ def generate_answer(text, image):
     return tokenizer.decode(output_ids[input_ids.shape[1]:], skip_special_tokens=True).strip()
 
 text_input = gr.Textbox(lines=5, label="Enter text")
-image_input = gr.Image(shape=(224, 224), label="Upload Image")
+image_input = gr.Image(type="pil", label="Upload Image")
 
 iface = gr.Interface(
     fn=generate_answer,