Update app.py

app.py

@@ -109,7 +109,7 @@ def update(sequence=DEFAULT_SEQ):
     with open(tmp.name, "w") as f:
         f.write(pdb_string)
 
-    return molecule(tmp.name), "test.jpg"
+    return molecule(tmp.name), tmp.name
 
 
 def suggest(option):
@@ -152,8 +152,9 @@ with demo:
         inp = gr.Textbox(label="Protein sequence", lines=3, value=DEFAULT_SEQ, placeholder="Write your protein sequence here...")
         with gr.Column():
             btn = gr.Button("🔬 Predict Structure ").style(full_width=False)
+            download = gr.File(label="Download file")
     mol = gr.HTML(update)    
-    download = gr.File(label="Download file")
+    #download = gr.File(label="Download file")
     btn.click(fn=update, inputs=inp, outputs=[mol, download])
     
     name.change(fn=suggest, inputs=name, outputs=inp)