use better transformers

app.py

@@ -1,9 +1,12 @@
 from pathlib import Path
 import gradio as gr
-from transformers import pipeline
+import torch
+# from transformers import pipeline
 from transformers import AutoModelForImageClassification
 import shutil
+from optimum.pipelines import pipeline
 
+device = 1 if torch.cuda.is_available() else "cpu"
 
 chk_point = "davanstrien/autotrain-encyclopaedia-illustrations-blog-post-3327992158"
 
@@ -12,6 +15,7 @@ model = AutoModelForImageClassification.from_pretrained(chk_point)
 pipe = pipeline(
     "image-classification",
     "davanstrien/autotrain-encyclopaedia-illustrations-blog-post-3327992158",
+    accelerator="bettertransformer",device=device
 )
 
 

requirements.txt

@@ -1 +1,2 @@
-transformers[torch]
+transformers[torch]
+optimum