Manojajj commited on
Commit
fb92761
·
verified ·
1 Parent(s): 96130bd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -0
app.py CHANGED
@@ -14,6 +14,11 @@ def authenticate_hf(token):
14
  except Exception as e:
15
  return f"Error: {e}"
16
 
 
 
 
 
 
17
  # Function to extract text from PDF
18
  def extract_text_from_pdf(pdf_path):
19
  with pdfplumber.open(pdf_path) as pdf:
 
14
  except Exception as e:
15
  return f"Error: {e}"
16
 
17
+ # Initialize the model and tokenizer
18
+ model_name = "meta-llama/Llama-3.1-70B-Instruct" # Replace with the actual model name
19
+ tokenizer = AutoTokenizer.from_pretrained(model_name)
20
+ model = AutoModelForCausalLM.from_pretrained(model_name)
21
+
22
  # Function to extract text from PDF
23
  def extract_text_from_pdf(pdf_path):
24
  with pdfplumber.open(pdf_path) as pdf: