Update use.py

use.py

@@ -1,7 +1,7 @@
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
-def run_pin_inference(prompt, model_id="LH-Tech-AI/Pin-Tiny", subfolder="Pin-25M"):
+def run_pin_inference(prompt, model_id="LH-Tech-AI/Pin-Tiny", subfolder="Pin-Ultra-25M"):
     # 1. Device Setup
     device = "cuda" if torch.cuda.is_available() else "cpu"
     print(f"Using device: {device}")
@@ -40,7 +40,7 @@ def run_pin_inference(prompt, model_id="LH-Tech-AI/Pin-Tiny", subfolder="Pin-25M
 # --- Sample test ---
 if __name__ == "__main__":
     user_query = "What is the weather like today?"
-    answer = run_pin_inference(user_query, model_id="LH-Tech-AI/Pin", subfolder="Pin-25M")
+    answer = run_pin_inference(user_query, model_id="LH-Tech-AI/Pin", subfolder="Pin-Ultra-25M")
     
     print(f"\nUser: {user_query}")
     print(f"Pin: {answer}")