Update app.py

app.py

@@ -11,7 +11,6 @@ Original file is located at
 #!pip install -Uq transformers datasets timm accelerate evaluate
 
 import gradio as gr
-from safetensors.torch import load_model
 from timm import create_model
 from huggingface_hub import hf_hub_download
 from datasets import load_dataset