Spaces:

dingusagar
/

aita-classifier

Sleeping

dingusagar commited on Nov 29, 2024

Commit

d542db8

verified ·

1 Parent(s): 332bb05

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -5,9 +5,9 @@ from ollama import chat
 from ollama import ChatResponse
 # Default model
-# OLLAMA_MODEL = "llama3.2:3b"
 # OLLAMA_MODEL = "llama3.2:1b"
-OLLAMA_MODEL = "mannix/qwen2-1.5b"
 # Load BERT MODEL
 from transformers import pipeline, DistilBertTokenizerFast

 from ollama import ChatResponse
 # Default model
+OLLAMA_MODEL = "llama3.2:3b"
 # OLLAMA_MODEL = "llama3.2:1b"
+# OLLAMA_MODEL = "mannix/qwen2-1.5b"
 # Load BERT MODEL
 from transformers import pipeline, DistilBertTokenizerFast