BasselAhmed commited on
Commit
ccd32c7
·
verified ·
1 Parent(s): 77bd5d0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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', 'seyonec/SMILES_tokenized_PubChem_shard00_160k',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):
 
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):