Hugging Face's logo

Spaces:
Hukuna
/
ProteinDesignDemo
Sleeping

App Files Community
Hukuna commited on
Commit
7ef2ddb
·
verified ·
1 Parent(s): 0caf69d

Update demo.py

Browse files
Files changed (1) hide show
  1. demo.py +1 -1
demo.py CHANGED
@@ -34,7 +34,7 @@ def download(filename,description):
34
  import pandas as pd
35
  def display(output,style,resn):
36
  # imformation
37
- protein=Protein.from_PDB(output)
38
  st.subheader("Protein Information:")
39
  st.write(f"Device: {protein.device}")
40
  st.write(f"Protein Length: {len(protein)} residues")
 
34
  import pandas as pd
35
  def display(output,style,resn):
36
  # imformation
37
+ protein=Protein.from_PDB(output,device=device)
38
  st.subheader("Protein Information:")
39
  st.write(f"Device: {protein.device}")
40
  st.write(f"Protein Length: {len(protein)} residues")