Update app.py

app.py

@@ -10,7 +10,7 @@ from safetensors.torch import load_file  # Import safetensors loading function
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
 
 # Define the directory containing the model files
-model_dir = "."  # Use current directory
+model_dir = "./models/best_model_"  # Adjusted to point to the best_model_ directory
 
 # Define paths to the specific model files
 model_path = os.path.join(model_dir, "model.safetensors")