Update app.py

app.py

@@ -5,9 +5,6 @@ import os
 import cv2
 from PIL import Image 
 import tensorflow as tf
-from transformers import AutoModel
-
-
 
 fp0 = np.zeros((96, 128), dtype = np.uint8)
 
@@ -136,8 +133,7 @@ def update_button(filepath, z):
     return filepath_show, [filepath], fp0, fp0
 
 def detect_cells(filepath, z):
-    #model = tf.keras.models.load_model('./model_positions', compile=False)
-    model = AutoModel.from_pretrained('./model_positions')
+    model = tf.keras.models.load_model('./model_positions', compile=False)
 
     img = tiff.imread(filepath)