Update app.py

app.py

@@ -17,7 +17,8 @@ def get_model_t5(model_name,tokenizer_name):
 
 
 def predict_gpt(text, model, tokenizer, temperature=1.0):
-    input_ids = tokenizer.encode(text+" \n Описание:", return_tensors="pt")
+    text = text.replace('\n','')+'Описание:'
+    input_ids = tokenizer.encode(text, return_tensors="pt")
 
     model.eval()
     with torch.no_grad():
@@ -67,7 +68,7 @@ def generate(model,temp,text):
     return result
          
 
-gpt_model, gpt_tokenizer = get_model_gpt('gpt/', 'gpt/')
+gpt_model, gpt_tokenizer = get_model_gpt('mipatov/rugpt3_nb_descr', 'mipatov/rugpt3_nb_descr')
 t5_model, t5_tokenizer = get_model_t5('mipatov/rut5_nb_descr', 'mipatov/rut5_nb_descr')