Update app.py

app.py

@@ -3,7 +3,6 @@ import os
 import streamlit as st
 import sys
 import urllib
-from transformers import GPT2Tokenizer, GPT2LMHeadModel
 import json
 import torch
 
@@ -11,7 +10,7 @@ def generate(tokenizer, model, text, features):
     generated = tokenizer("<|startoftext|> <|titlestart|>{}<|titleend|>".format(text), return_tensors="pt").input_ids
     sample_outputs = model.generate(
         generated, do_sample=True, top_k=50, 
-        max_length=300, top_p=0.95, temperature=2.1, num_return_sequences=2
+        max_length=300, top_p=features['top_p'], temperature=features['t'] / 100.0, num_return_sequences=features['num'],
     )
     for i, sample_output in enumerate(sample_outputs):
         decoded = tokenizer.decode(sample_output, skip_special_tokens=True).split(text)[1]
@@ -19,8 +18,8 @@ def generate(tokenizer, model, text, features):
 
 
 def load_model():
-    tokenizer = torch.load('./tokenizer.pt')
-    model = torch.load('./model.pt', map_location=torch.device('cpu'))
+    tokenizer = torch.load('./final_model/tokenizer.pt')
+    model = torch.load('./final_model/model.pt', map_location=torch.device('cpu'))
     return tokenizer, model
 
 
@@ -71,4 +70,4 @@ def main():
     generate(tokenizer, model, data['title'], features)
 
 if __name__ == "__main__":
-    main()
+    main()