Update app_flash1.py

app_flash1.py

@@ -128,10 +128,11 @@ def get_flashpack_model(hf_repo="rahul7star/FlashPack"):
 
     local_path = "model.flashpack"
 
-    # 1️⃣ Check if local copy already exists
+    # 1️⃣ Try local first
     if os.path.exists(local_path):
         print("✅ Found local model.flashpack — loading it directly.")
         model = GemmaTrainer().from_flashpack(local_path)
+        model.eval()
         tokenizer, embed_model, _ = build_encoder("gpt2")
         print("✅ FlashPack model loaded successfully (local).")
         return model, tokenizer, embed_model, None, None
@@ -143,7 +144,7 @@ def get_flashpack_model(hf_repo="rahul7star/FlashPack"):
             print("✅ Found model.flashpack in repo — downloading and loading it.")
             local_path = hf_hub_download(repo_id=hf_repo, filename="model.flashpack")
 
-            # Load cleanly using the class itself
+            # Correct loading: no extra args passed
             model = GemmaTrainer().from_flashpack(local_path)
             model.eval()