ashok2216 commited on
Commit
9d423d7
1 Parent(s): 7097291

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -3,8 +3,8 @@ import streamlit as st
3
  from transformers import GPT2LMHeadModel, GPT2Tokenizer
4
 
5
  # Load pre-trained GPT-3 model and tokenizer
6
- tokenizer = GPT2Tokenizer.from_pretrained("meta-llama/Meta-Llama-3-8B")
7
- model = GPT2LMHeadModel.from_pretrained("meta-llama/Meta-Llama-3-8B")
8
 
9
  def extract_text_from_pdf(pdf_path):
10
  text = ""
 
3
  from transformers import GPT2LMHeadModel, GPT2Tokenizer
4
 
5
  # Load pre-trained GPT-3 model and tokenizer
6
+ tokenizer = GPT2Tokenizer.from_pretrained("openai-community/gpt2")
7
+ model = GPT2LMHeadModel.from_pretrained("openai-community/gpt2")
8
 
9
  def extract_text_from_pdf(pdf_path):
10
  text = ""