Update app.py

app.py

@@ -85,43 +85,7 @@ def paraphrase(sentence: str, count: str):
         return {'result': []}
     sentence_input = sentence
     text = f"paraphrase: {sentence_input} </s>"
-    # encoding = tokenizer.encode_plus(text, padding=True, return_tensors="pt")
     encoding = tokenizer(text, return_tensors="pt")
-    # input_ids, attention_masks = encoding["input_ids"], encoding["attention_mask"]
-    
-    # outputs = model.generate(
-    #     input_ids=input_ids, attention_mask=attention_masks,
-    #     max_length=512,  # 256,
-    #     do_sample=True,
-    #     top_k=120,
-    #     top_p=0.95,
-    #     early_stopping=True,
-    #     num_return_sequences=p_count
-    # )
-    # res = []
-    # for output in outputs:
-    #     line = tokenizer.decode(
-    #         output, skip_special_tokens=True, clean_up_tokenization_spaces=True)
-    #     res.append(line)
-
-    # print(res)
-
-    # 
-    # input_ids, attention_masks = encoding["input_ids"], encoding["attention_mask"]
-    # outputs = model.generate(input_ids=input_ids, 
-    #                          attention_mask=attention_masks,
-    #                          max_length=512,
-    #                          do_sample=True,
-    #                          top_k=120,
-    #                          top_p=0.95,
-    #                          early_stopping=True,
-    #                          num_return_sequences=p_count)
-    # result_v4 = []
-    # for output in outputs:
-    #     line = tokenizer.decode(output, skip_special_tokens=True, clean_up_tokenization_spaces=True)
-    #     result_v4.append(line)
-    # 
-    
     return {
         'result': {
             'generate_v1':generate_v1(encoding, p_count),
@@ -130,7 +94,8 @@ def paraphrase(sentence: str, count: str):
             'generate_v4':generate_v4(encoding, p_count)
         }
     }
-        
+
+
 def paraphrase_dummy(sentence: str, count: str):
     return {'result': []}