Spaces:
Running
on
Zero
Running
on
Zero
vladimir.manuylov
commited on
Commit
·
6d736db
1
Parent(s):
58ac1b4
fixing bug
Browse files
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 ---
|