Update app.py

app.py

@@ -1,9 +1,15 @@
 import subprocess
+import sys
 
-# Find where mednextv1 is installed
-subprocess.run(["find", "/usr/local/lib/python3.10/site-packages", "-name", "mednextv1"])
-
+print("Installing mednextv1...")
+subprocess.run([sys.executable, "-m", "pip", "install", "--force-reinstall", "git+https://github.com/FrancescoLR/MedNeXt.git"])
 
+try:
+    from mednextv1 import create_mednext_encoder_v1
+    print("Successfully imported mednextv1!")
+except ModuleNotFoundError:
+    print("ERROR: mednextv1 still not found after reinstalling!")
+    
 
 import gradio as gr
 import torch