Spaces:

iteratehack
/

code-19

Sleeping

PiotrPasztor commited on 14 days ago

Commit

176390f

1 Parent(s): 4b91e03

fix

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,7 +7,7 @@ from pydantic import BaseModel
 from transformers import AutoTokenizer, AutoModel
 # Simple RL Classifier using Transformer
-ACTIONS = ["TRIP", "GITHUB", "MESSAGE"]
 DATASET_PATH = os.path.join(os.path.dirname(__file__), "dataset.jsonl")
 app = FastAPI()

 from transformers import AutoTokenizer, AutoModel
 # Simple RL Classifier using Transformer
+ACTIONS = ["TRIP", "NONE", "GITHUB", "MAIL"]
 DATASET_PATH = os.path.join(os.path.dirname(__file__), "dataset.jsonl")
 app = FastAPI()