Ctaake commited on
Commit
c436cce
·
verified ·
1 Parent(s): 3c35121

Switch to gemma

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ from datetime import datetime
9
  # Model which is used
10
  checkpoint = "CohereForAI/c4ai-command-r-v01"
11
  checkpoint = "mistralai/Mistral-7B-Instruct-v0.1"
12
- #checkpoint = "google/flan-t5-small"
13
  path_to_log = "FlaggedFalse.txt"
14
 
15
  # Inference client with the model (And HF-token if needed)
 
9
  # Model which is used
10
  checkpoint = "CohereForAI/c4ai-command-r-v01"
11
  checkpoint = "mistralai/Mistral-7B-Instruct-v0.1"
12
+ checkpoint = "google/gemma-1.1-7b-it"
13
  path_to_log = "FlaggedFalse.txt"
14
 
15
  # Inference client with the model (And HF-token if needed)