skylersterling commited on
Commit
22fb6e4
·
verified ·
1 Parent(s): db9d4db

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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):