Hugging Face's logo

Spaces:
TechDisciples1
/
Theo
Sleeping

App Files Community
alaselababatunde commited on
Commit
254e2bc
·
1 Parent(s): 2296ef1

Updated

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -20,7 +20,7 @@ from transformers import pipeline
20
  # CONFIGURATION
21
  # =====================================================
22
  API_SECRET = "techdisciplesai404"
23
- PRIMARY_MODEL = "christopherastone/distilgpt2-proofs"
24
  FALLBACK_MODEL = "mistralai/Mistral-7B-Instruct-v0.3"
25
  DEVICE = 0 if torch.cuda.is_available() else -1
26
 
 
20
  # CONFIGURATION
21
  # =====================================================
22
  API_SECRET = "techdisciplesai404"
23
+ PRIMARY_MODEL = "Emmet-Allen/christAIn"
24
  FALLBACK_MODEL = "mistralai/Mistral-7B-Instruct-v0.3"
25
  DEVICE = 0 if torch.cuda.is_available() else -1
26