Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -3,7 +3,7 @@ from transformers import pipeline, AutoTokenizer
|
|
3 |
import base64
|
4 |
|
5 |
# Load the GPT-2 text generation model from Hugging Face
|
6 |
-
text_generator = pipeline("text-generation", model="
|
7 |
|
8 |
# Increase the maximum token limit
|
9 |
tokenizer = AutoTokenizer.from_pretrained("EleutherAI/gpt-neo-2.7B")
|
|
|
3 |
import base64
|
4 |
|
5 |
# Load the GPT-2 text generation model from Hugging Face
|
6 |
+
text_generator = pipeline("text-generation", model="Shruti9756/G24_Contract_Summarization_step3")
|
7 |
|
8 |
# Increase the maximum token limit
|
9 |
tokenizer = AutoTokenizer.from_pretrained("EleutherAI/gpt-neo-2.7B")
|