New viz

app.py

@@ -16,7 +16,7 @@ import warnings
 from datetime import datetime
 from pathlib import Path
 BASE_DIR = Path(__file__).resolve().parent
-LOCAL_COMPONENT_PATH = BASE_DIR / "gradio_molecule3d" / "backend"
+LOCAL_COMPONENT_PATH = BASE_DIR / "gradio_molecule3d_2" / "backend"
 sys.path.insert(0, str(LOCAL_COMPONENT_PATH))
 from gradio_molecule3d.molecule3d import Molecule3D
 from Bio.PDB import PDBParser, PDBIO
@@ -269,14 +269,7 @@ def flex_3d(input_file):
 
     model.to(config['inference_args']['device'])
     print("Loading 3D model from {}".format(config['inference_args']['3d_model_path']))
-    repo_id = "Honzus24/Flexpert_weights"
-    file_weights = config['inference_args']['3d_model_path']
-    
-    # Get path (instant if cached)
-    weights_path = get_weights_path(repo_id, file_weights)
-    
-    # Load weights
-    state_dict = torch.load(weights_path, map_location=config['inference_args']['device'])
+    state_dict = torch.load(config['inference_args']['3d_model_path'], map_location=config['inference_args']['device'])
     model.load_state_dict(state_dict, strict=False)
     model.eval()
 
@@ -634,4 +627,4 @@ with gr.Blocks(theme=theme) as demo:
     predict_3d.click(handle_3d_prediction, inputs=[input_file_3d], outputs=[output_files, output_text, molecule_output])
 
 # Launch the interface
-demo.launch(show_error=True, share=True)
+demo.launch(show_error=True)