Update main.py

main.py

@@ -145,7 +145,7 @@ def FREEGPT(
         stream : bool = True,
         timeout: Optional[float] = None
 ):
-    md = next((item["tag"] + "/" + item["model"] for item in MODELS if item["model"] == model), "@cf/meta/llama-3.2-1b-instruct")
+    md = next((item["tag"] + "/" + item["model"] for item in M2 if item["model"] == model), "@cf/meta/llama-3.2-1b-instruct")
 
     URL = f"https://llmchat.in/inference/stream?model={md}"
 
@@ -166,7 +166,7 @@ def FREEGPT(
     "messages": messages,
     "stream": stream,
     **({"max_tokens": max_token} if max_token is not None else {}),
-    **({"max_tokens": next((item["max_tokens"] for item in MODELS if item["model"] == model and item["max_tokens"] is not None), None)} if next((True for item in MODELS if item["model"] == model and item["max_tokens"] is not None), None) else {})
+    **({"max_tokens": next((item["max_tokens"] for item in M2 if item["model"] == model and item["max_tokens"] is not None), None)} if next((True for item in M2 if item["model"] == model and item["max_tokens"] is not None), None) else {})
 }
 
 
@@ -228,9 +228,9 @@ def QWEN(
         
         headers = {"Accept": "*/*","Content-Type": "application/json","Origin": "https://teichai-qwen3-4b-thinking-2507-claude-4-5-opus.hf.space","Referer": "https://teichai-qwen3-4b-thinking-2507-claude-4-5-opus.hf.space/","User-Agent": "python-requests/2.x"}
 
-        c = t()
+
         RESPO = RQ.post(API_URL, headers=headers, json=payload, stream=stream, timeout=timeout)
-        print(c-t())
+
         # print(RESPO)
         buffer_lines = []
         for raw in RESPO.iter_lines():
@@ -675,4 +675,4 @@ def info():
 # ---------------------------------------------------------------------
 
 # if __name__ == "__main__":
-#     app.run(host="0.0.0.0", port=5550, threaded=True)
+#     app.run(host="0.0.0.0", port=5550, threaded=True)