update app

app.py

@@ -3,6 +3,9 @@ import gradio as gr
 import requests
 import torch
 from PIL import Image
+import timm
+from timm.data import resolve_data_config
+from timm.data.transforms_factory import create_transform
 
 rationale_model_dir = "cooelf/MM-CoT-UnifiedQA-Base-Rationale-Joint"
 vit_model = timm.create_model("vit_base_patch16_384", pretrained=True, num_classes=0)