Update app.py

app.py

@@ -5,14 +5,16 @@ import torch
 
 #Qwen/Qwen2.5-14B-Instruct-1M
 #Qwen/Qwen2-0.5B
-model_name = "Qwen/Qwen2-0.5B"
+model_name = "bobber/Qwen-0.5B-GRPO"
+subfolder = "Qwen-0.5B-GRPO/checkpoint-1868"
 
 model = AutoModelForCausalLM.from_pretrained(
     model_name,
+    subfolder=subfolder,
     torch_dtype=torch.bfloat16,
     device_map="auto"
 )
-tokenizer = AutoTokenizer.from_pretrained(model_name)
+tokenizer = AutoTokenizer.from_pretrained(model_name, subfolder=subfolder)
 SYSTEM_PROMPT = """
 Respond in the following format:
 <reasoning>