Spaces:

Sid26Roy
/

query-type

Sleeping

Sid26Roy commited on Jun 26

Commit

bd2f52d

verified ·

1 Parent(s): e52cc89

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -12,7 +12,7 @@ model.to(device)
 model.eval()
 # ID to label mapping
-id2label = {0: "Select", 1: "Insert", 2: "Delete", 3: "Update", 4: "Analyse"}
 def classify_query(text):
     if not text.strip():

 model.eval()
 # ID to label mapping
+id2label = {0: "Update", 1: "Analyse", 2: "Insert", 3: "Delete", 4: "Select"}
 def classify_query(text):
     if not text.strip():