Update app.py

app.py

@@ -1,6 +1,6 @@
 import torch
 import gradio as gr
-from transformers import pipeline
+from optimum.pipelines import pipeline
 import ast
 
 translation_task_names = {
@@ -58,6 +58,7 @@ def translate_text(model_choice, task_choice, text_input, load_in_8bit, device):
                             device=device,  # Use selected device
                             model_kwargs=model_kwargs, 
                             torch_dtype=dtype,  # Set the floating point
+                            accelerator="bettertransformer", # Use optimum bettertransformer
                             use_fast=True
                             )
         # Store the loaded model