Hugging Face's logo

Spaces:
TechDisciples1
/
Theo
Sleeping

App Files Community
alaselababatunde commited on
Commit
328d04d
·
1 Parent(s): 191212a

Updated

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