Update app.py

app.py

@@ -28,6 +28,7 @@ import pandas as pd
 from PIL import Image
 import gradio
 from functools import partial
+from transformers import AutoModel
 
 """### Set parameters"""