Update app.py

app.py

@@ -1,7 +1,9 @@
-import tensorflow as tf
-from tensorflow import keras
 import numpy as np
 import gradio as gr
+# teo
+from transformers import Dinov2Config, Dinov2Model, Dinov2ForImageClassification, AutoImageProcessor
+import torch.nn as nn
+import os
 
 # Load PyTorch model
 def create_head(num_features , number_classes ,dropout_prob=0.5 ,activation_func =nn.ReLU):
@@ -14,7 +16,6 @@ def create_head(num_features , number_classes ,dropout_prob=0.5 ,activation_func
         if dropout_prob !=0 : layers.append(nn.Dropout(dropout_prob))
     layers.append(nn.Linear(features_lst[-1] , number_classes))
     return nn.Sequential(*layers)
-from transformers import Dinov2Config, Dinov2Model
 
 class NewheadDinov2ForImageClassification(Dinov2ForImageClassification):
     def __init__(self, config: Dinov2Config) -> None: