rwitz commited on
Commit
c2022d2
1 Parent(s): 3ce557c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -22,7 +22,7 @@ def classify_vote(user_input):
22
  payload = {
23
  "inputs": user_input,
24
  "parameters": {
25
- "candidate_labels": "roleplay,conversation,mathematics,logic,creative writing,knowledge",
26
  "multi_class": True
27
  }
28
  }
@@ -51,7 +51,7 @@ password=os.environ.get("MONGODB")
51
  def init_database():
52
  uri = f"mongodb+srv://new-user:{password}@cluster0.xb2urf6.mongodb.net/?retryWrites=true&w=majority&appName=Cluster0"
53
  client = MongoClient(uri)
54
- db = client["elo_ratings3"]
55
  collection = db["elo_ratings"]
56
  return collection
57
 
 
22
  payload = {
23
  "inputs": user_input,
24
  "parameters": {
25
+ "candidate_labels": "roleplay,conversation,pure mathematics,logic problems,creative writing,knowledge",
26
  "multi_class": True
27
  }
28
  }
 
51
  def init_database():
52
  uri = f"mongodb+srv://new-user:{password}@cluster0.xb2urf6.mongodb.net/?retryWrites=true&w=majority&appName=Cluster0"
53
  client = MongoClient(uri)
54
+ db = client["elo_ratings4"]
55
  collection = db["elo_ratings"]
56
  return collection
57