Update app.py
Browse files
app.py
CHANGED
|
@@ -27,7 +27,7 @@ for pkg in pkgs: print(pkg)
|
|
| 27 |
class Query(BaseModel):
|
| 28 |
smiles :List[str]
|
| 29 |
|
| 30 |
-
rob_chem_model = ClassificationModel('roberta', '
|
| 31 |
|
| 32 |
@app.post("/ToxicityPrediction")
|
| 33 |
def c(query:Query):
|
|
|
|
| 27 |
class Query(BaseModel):
|
| 28 |
smiles :List[str]
|
| 29 |
|
| 30 |
+
rob_chem_model = ClassificationModel('roberta', 'BasselAhmed/RobertaChemClinToxTuned',use_cuda=False ,args={'evaluate_each_epoch':True , 'evaluate_during_training_verbose':True, 'seed':4})
|
| 31 |
|
| 32 |
@app.post("/ToxicityPrediction")
|
| 33 |
def c(query:Query):
|