Update app.py

app.py

@@ -37,7 +37,7 @@ def run_gpt(
     purpose,
     **prompt_kwargs,
 ):
-    seed = random.randint(1,1111111111111111)
+    seed = random.randint(1, 1111111111111111)
     print(seed)
     generate_kwargs = dict(
         temperature=1.0,
@@ -53,11 +53,12 @@ def run_gpt(
         purpose=purpose,
         safe_search=safe_search,
     ) + prompt_template.format(**prompt_kwargs)
+    
     if True:
         print(LOG_PROMPT.format(content))
 
-    model = pipeline('text-generation', model='microsoft/DialoGPT-small')
-    response = model(content, max_length=max_tokens, temperature=1.0)
+    model = pipeline('text-generation', model='microsoft/DialoGPT-small', pad_token_id=model.config.eos_token_id)  # Set pad_token_id
+    response = model(content, max_length=max_tokens, temperature=1.0, truncation=True)  # Explicitly set truncation
     resp = response[0]['generated_text']
 
     if True: