Update app.py

app.py

@@ -5,6 +5,9 @@ import os
 import cv2
 from PIL import Image 
 import tensorflow as tf
+from transformers import AutoModel
+
+
 
 fp0 = np.zeros((96, 128), dtype = np.uint8)
 
@@ -133,7 +136,8 @@ 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 = tf.keras.models.load_model('./model_positions', compile=False)
+    model = AutoModel.from_pretrained('./model_positions')
 
     img = tiff.imread(filepath)