Upload utils.py

utils.py

@@ -44,8 +44,8 @@ def get_t5_model():
 def get_flan_t5_model():
     return pipeline(
         "summarization",
-        model="google/flan-t5-base",
-        tokenizer="google/flan-t5-base",
+        model="google/flan-t5-small",
+        tokenizer="google/flan-t5-small",
         max_length=512,
         # length_penalty = 0
     )
@@ -375,11 +375,8 @@ def gpt_model(prompt):
     response = openai.Completion.create(
         model="text-davinci-003",
         prompt=prompt,
-        temperature=0.1,
+        temperature=0,
         max_tokens=1024,
-        top_p=1.0,
-        frequency_penalty=0.5,
-        presence_penalty=1,
     )
     return response.choices[0].text