bravewiki commited on
Commit
ea12b72
·
verified ·
1 Parent(s): caf9575

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from transformers import pipeline
4
  from PIL import Image
5
 
6
  # Load Hugging Face model
7
- model = pipeline("text-generation", model="meta-llama/Llama-3.2-1B")
8
 
9
  def extract_text_from_pdf(pdf_file):
10
  reader = PdfReader(pdf_file)
 
4
  from PIL import Image
5
 
6
  # Load Hugging Face model
7
+ model = pipeline("text-generation", model="Ketak-ZoomRx/Drug_Prompt_Ollama_67k")
8
 
9
  def extract_text_from_pdf(pdf_file):
10
  reader = PdfReader(pdf_file)