Update app.py

app.py

@@ -11,12 +11,20 @@ chk_point = "davanstrien/autotrain-ia-useful-covers-3665397856"
 
 model = AutoModelForImageClassification.from_pretrained(chk_point)
 
-pipe = pipeline(
+try:
+    pipe = pipeline(
     "image-classification",
     chk_point,
     accelerator="bettertransformer",device=device
 )
-
+except NotImplementedError:
+    from transformers import pipeline
+    pipe = pipeline(
+    "image-classification",
+    chk_point,
+    device=device
+)
+    
 
 def make_label_folders():
     folders = model.config.label2id.keys()