Update app.py

app.py

@@ -1,11 +1,13 @@
 import gradio as gr
-from transformers import ViTFeatureExtractor, ViTModel
 import torch
-import matplotlib.pyplot as plt
+from transformers import ViTFeatureExtractor, ViTModel
 from skops import hub_utils
 from einops import reduce
+import matplotlib.pyplot as plt
 import seaborn as sns
+
 import pickle
+import os
 
 labels = [
     'tench',
@@ -26,7 +28,7 @@ feature_extractor = ViTFeatureExtractor.from_pretrained('facebook/dino-vitb16')
 model = ViTModel.from_pretrained('facebook/dino-vitb16').eval().to(device)
 
 # load logistic regression
-!mkdir emb-gam-dino
+os.mkdir('emb-gam-dino')
 hub_utils.download(repo_id='Ramos-Ramos/emb-gam-dino', dst='emb-gam-dino')
 
 with open('emb-gam-dino/model.pkl', 'rb') as file: