Update app.py

app.py

@@ -37,9 +37,12 @@ print("="*60)
 
 try:
     # Check if trained model files exist in root
-    model_files = ['config.json', 'pytorch_model.bin', 'sentence_bert_config.json']
-    has_model = all(os.path.exists(f) for f in model_files)
-    
+    # Check if trained model files exist in root
+    model_files = ['config.json', 'sentence_bert_config.json']
+    # Check for either pytorch_model.bin or model.safetensors
+    has_weights = os.path.exists('pytorch_model.bin') or os.path.exists('model.safetensors')
+    has_model = all(os.path.exists(f) for f in model_files) and has_weights
+        
     if has_model:
         print("✓ Trained model files found in root directory")
         print("Loading trained model...")