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

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-classification", model="medicalai/ClinicalBERT")
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="meta-llama/Llama-3.2-1B")
8
 
9
  def extract_text_from_pdf(pdf_file):
10
  reader = PdfReader(pdf_file)