Update app.py

app.py

@@ -4,7 +4,7 @@ from transformers import pipeline
 from PIL import Image
 
 # Load Hugging Face model
-model = pipeline("text-generation", model="Ketak-ZoomRx/Drug_Prompt_Ollama_67k")
+model = pipeline("text-generation", model="Ketak-ZoomRx/Drug_Prompt_Ollama_67k",torch_dtype="auto", trust_remote_code=True, use_fast=True, device_map={"": "cuda:0"},)
 
 def extract_text_from_pdf(pdf_file):
     reader = PdfReader(pdf_file)