mod app v2

app.py

@@ -207,16 +207,16 @@ class LanguageModel(nn.Module):
 
 # Load the model
 model = LanguageModel().to(device)
-model_path = "model_v6_flash_attn.pth"
+model_path = "./model_v6_flash_attn.pth"
 
 # Check if model file exists
 if os.path.exists(model_path):
-    model.load_state_dict(torch.load(model_path, map_location=device))
+    model.load_state_dict(torch.load(
+        model_path, map_location=device, weights_only=False))
     model.eval()
     print("Model loaded successfully")
 else:
-    print(f"Model file not found at {
-          model_path}. Please train the model first.")
+    print("model file not found")
 
 # Compile model for better performance
 model = torch.compile(model)