Update app.py

app.py

@@ -11,7 +11,7 @@ def get_pdb(upload_choice="PDB Code", pdb_code="", filepath=""):
         "AlphaFold DB": {"base":"https://alphafold.ebi.ac.uk/files/", "suffix": "-F1-model_v4.pdb"}, 
         "ESM Atlas": {"base": "https://api.esmatlas.com/fetchPredictedStructure/", "suffix":".pdb"}
     }
-    if pdb_code is None or pdb_code == "":
+    if upload_choice=="local file":
         try:
             #move file to home folder to have it accessible from the web
             copyfile(filepath.name, os.path.join(os.getcwd(), os.path.basename(filepath.name)))
@@ -141,7 +141,7 @@ with demo:
 
             
             btn = gr.Button("View structure")
-    gr.Examples([["PDB Code", "2CBA"], ["ESM Atlas", "MGYP001531319262"]], [upload_choice,inp])
+    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