Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -9,6 +9,7 @@ HF_TOKEN = os.environ.get("HF_TOKEN")
|
|
9 |
# Load the tokenizer and model
|
10 |
tokenizer = GPT2Tokenizer.from_pretrained('skylersterling/TopicGPT', use_auth_token=HF_TOKEN)
|
11 |
model = GPT2LMHeadModel.from_pretrained('skylersterling/TopicGPT', use_auth_token=HF_TOKEN)
|
|
|
12 |
|
13 |
# Define the function that generates text from a prompt
|
14 |
def generate_text(prompt):
|
|
|
9 |
# Load the tokenizer and model
|
10 |
tokenizer = GPT2Tokenizer.from_pretrained('skylersterling/TopicGPT', use_auth_token=HF_TOKEN)
|
11 |
model = GPT2LMHeadModel.from_pretrained('skylersterling/TopicGPT', use_auth_token=HF_TOKEN)
|
12 |
+
model.eval()
|
13 |
|
14 |
# Define the function that generates text from a prompt
|
15 |
def generate_text(prompt):
|