Update app.py

app.py

@@ -1,16 +1,18 @@
-import numpy as np
+from typing import Optional
 import os
+os.environ["WANDB_DISABLED"] = "true"
+
+import numpy as np
+
+from PIL import Image
 import gradio as gr
 import torch
 import torch.nn as nn
-from PIL import Image
 from transformers import CLIPModel, AutoModel
-from typing import Optional
 
+from huggingface_hub import hf_hub_download
 from safetensors.torch import load_model
 
-os.environ["WANDB_DISABLED"] = "true"
-
 from datasets import load_dataset, load_metric
 from transformers import (
     AutoConfig,