Update main.py

main.py

@@ -5,7 +5,7 @@ import requests
 import json
 
 # 处理 API 请求的函数
-def api_call(input_text):
+def api_call(input_text, model="internlm2.5-latest", temperature=0.8, top_p=0.9, n=1):
     api_token = os.getenv("API_TOKEN")
     if not api_token:
         return "API token 未设置，请检查环境变量。"
@@ -45,11 +45,11 @@ def api_call(input_text):
     }
     
     data = {
-        "model": "internlm2.5-latest",
+        "model": model,
         "messages": [{"role": "user", "content": input_text}],
-        "n": 1,
-        "temperature": 0.8,
-        "top_p": 0.9
+        "n": n,
+        "temperature": temperature,
+        "top_p": top_p
     }
     
     try:
@@ -85,7 +85,7 @@ def start_gradio():
         
         # 定义按钮点击事件
         send_button.click(
-            api_call_with_params,
+            api_call,
             inputs=[input_box, model_box, temperature_slider, top_p_slider, n_slider],
             outputs=output_box
         )