Hugging Face's logo

Spaces:
dingusagar
/
aita-classifier
Sleeping

App Files Community
dingusagar commited on
Commit
edb9001
·
verified ·
1 Parent(s): d542db8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -7,7 +7,7 @@ from ollama import ChatResponse
7
  # Default model
8
  OLLAMA_MODEL = "llama3.2:3b"
9
  # OLLAMA_MODEL = "llama3.2:1b"
10
- # OLLAMA_MODEL = "mannix/qwen2-1.5b"
11
 
12
  # Load BERT MODEL
13
  from transformers import pipeline, DistilBertTokenizerFast
@@ -25,8 +25,8 @@ classifier = pipeline(
25
  )
26
 
27
  bert_label_map = {
28
- 'LABEL_0': 'You are The A-hole (YTA)',
29
- 'LABEL_1': 'Not the A-hole (NTA)',
30
  }
31
 
32
  def ask_bert(prompt):
 
7
  # Default model
8
  OLLAMA_MODEL = "llama3.2:3b"
9
  # OLLAMA_MODEL = "llama3.2:1b"
10
+ OLLAMA_MODEL = "llama3.2:3b-instruct-q2_K"
11
 
12
  # Load BERT MODEL
13
  from transformers import pipeline, DistilBertTokenizerFast
 
25
  )
26
 
27
  bert_label_map = {
28
+ 'LABEL_0': 'YTA',
29
+ 'LABEL_1': 'NTA',
30
  }
31
 
32
  def ask_bert(prompt):