Update app.py

app.py

@@ -55,7 +55,6 @@ def get_model_pipeline(model_name):
         model=model,
         tokenizer=tokenizer,
         torch_dtype=torch.bfloat16,
-        device="cuda"
     )
     
     return pipe, model