Spaces:
Running
Running
Update app.py
Browse files
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("
|
7 |
-
model = GPT2LMHeadModel.from_pretrained("
|
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 = ""
|