vladimir.manuylov commited on
Commit
6d736db
·
1 Parent(s): 58ac1b4

fixing bug

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -18,6 +18,7 @@ from huggingface_hub import CommitScheduler
18
  REPO_ID = "ai-gero/ProtoBind-Diff"
19
  MODEL_FILENAME = "model.ckpt"
20
  TOKENIZER_FILENAME = "tokenizer_smiles_diffusion.json"
 
21
 
22
  @spaces.GPU(duration=120)
23
  def generate_smiles_for_sequence(protein_sequence: str, num_samples: int, request: Request):
@@ -78,7 +79,7 @@ scheduler = CommitScheduler(
78
  repo_type="dataset",
79
  folder_path=str(LOG_FOLDER),
80
  every=300,
81
- token=os.getenv("HF_TOKEN")
82
  )
83
 
84
  # --- GRADIO APP DEFINITION ---
 
18
  REPO_ID = "ai-gero/ProtoBind-Diff"
19
  MODEL_FILENAME = "model.ckpt"
20
  TOKENIZER_FILENAME = "tokenizer_smiles_diffusion.json"
21
+ Request = gr.Request
22
 
23
  @spaces.GPU(duration=120)
24
  def generate_smiles_for_sequence(protein_sequence: str, num_samples: int, request: Request):
 
79
  repo_type="dataset",
80
  folder_path=str(LOG_FOLDER),
81
  every=300,
82
+ token=os.getenv("HF_TOKEN")
83
  )
84
 
85
  # --- GRADIO APP DEFINITION ---