Upload external_api_engine.py

external_api_engine.py

@@ -3,8 +3,9 @@ import requests
 import os
 from emotion_filter import rewrite_with_persona
 
+# NIMOモデル（指示応答型）
+API_URL = "https://api-inference.huggingface.co/models/rinna/japanese-gpt-neox-3.6b-instruction-ppo"
 API_TOKEN = os.getenv("neko")
-API_URL = "https://api-inference.huggingface.co/models/rinna/japanese-gpt-neox-3.6b"
 
 headers = {
     "Authorization": f"Bearer {API_TOKEN}"
@@ -15,9 +16,10 @@ def call_external_api(input_text):
         payload = {
             "inputs": input_text,
             "parameters": {
-                "max_new_tokens": 100,
+                "max_new_tokens": 160,
                 "temperature": 0.8,
-                "do_sample": True
+                "do_sample": True,
+                "repetition_penalty": 1.2
             }
         }
         response = requests.post(API_URL, headers=headers, json=payload, timeout=30)