Update app.py

app.py

@@ -2,13 +2,18 @@ import gradio as gr
 from shutil import copyfile
 
 
+## Modify to match format username-spacename
+## Only works for hosted spaces on Huggingface, for local spaces or spaces hosted on Colab you need to use the gradio.live url with share=True as described in the last two lines
+public_link = "https://simonduerr-molstar-gradio.hf.space"
+
+
 import gradio as gr
 import os
 
 def get_pdb(upload_choice="PDB Code", pdb_code="", filepath=""):
     urls = {
         "PDB Code": {"base":"https://files.rcsb.org/view/", "suffix":".pdb"}, 
-        "AlphaFold DB": {"base":"https://alphafold.ebi.ac.uk/files/", "suffix": "-F1-model_v4.pdb"}, 
+        "AlphaFold DB": {"base":"https://alphafold.ebi.ac.uk/files/AF-", "suffix": "-F1-model_v4.pdb"}, 
         "ESM Atlas": {"base": "https://api.esmatlas.com/fetchPredictedStructure/", "suffix":".pdb"}
     }
     if upload_choice=="local file":
@@ -126,7 +131,7 @@ with demo:
     gr.Markdown("# PDB viewer using Mol*")
     gr.Markdown("""If using please cite 
     > David Sehnal, Sebastian Bittrich, Mandar Deshpande, Radka Svobodová, Karel Berka, Václav Bazgier, Sameer Velankar, Stephen K Burley, Jaroslav Koča, Alexander S Rose: Mol* Viewer: modern web app for 3D visualization and analysis of large biomolecular structures, Nucleic Acids Research, 2021; 10.1093/nar/gkab31.""")
-    public_link = gr.Variable(value="https://simonduerr-molstar-gradio.hf.space")
+    public_link = gr.Variable(value=public_link)
     with gr.Row():
         with gr.Box():
             upload_choice = gr.Radio(["PDB Code", "AlphaFold DB", "ESM Atlas","local file"], label="File source", value='PDB Code')
@@ -144,4 +149,5 @@ with demo:
     gr.Examples([["PDB Code", "2CBA"],["AlphaFold DB", "A0A1U8FD60"], ["ESM Atlas", "MGYP001531319262"]], [upload_choice,inp])
     mol = gr.HTML()
     btn.click(fn=update, inputs=[upload_choice, inp, file, public_link], outputs=mol)
-_, _, pl = demo.launch()  # use public link with share=True locally
+_, _, pl = demo.launch()  # use public link with share=True locally and uncomment below
+#public_link = pl