Upload inference.py with huggingface_hub

inference.py

@@ -25,7 +25,7 @@ from openai import OpenAI
 
 API_BASE_URL = os.getenv("API_BASE_URL", "https://router.huggingface.co/v1")
 API_KEY      = os.getenv("HF_TOKEN") or os.getenv("API_KEY", "")
-MODEL_NAME   = os.getenv("MODEL_NAME", "meta-llama/Llama-3.3-70B-Instruct")
+MODEL_NAME   = os.getenv("MODEL_NAME", "meta-llama/Llama-3.2-3B-Instruct")
 
 MAX_STEPS   = 15
 TEMPERATURE = 0.1
@@ -131,7 +131,7 @@ def run_task(
 
     # Reset
     result     = env.reset()
-    obs        = result.observation.dict()
+    obs        = result.observation.model_dump()
     done       = result.done
     step       = 0
     rewards    = []
@@ -177,7 +177,7 @@ def run_task(
 
         # Step environment
         result  = env.step(action)
-        obs     = result.observation.dict()
+        obs     = result.observation.model_dump()
         done    = result.done
         reward  = result.reward.total
         rewards.append(reward)