Hugging Face's logo

Spaces:
MASSJ77
/
flan-t5-chatbot
Sleeping

App Files Community
MASSJ77 commited on
Commit
3d2e39d
·
verified ·
1 Parent(s): f1c7fe1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ os.makedirs(CACHE_DIR, exist_ok=True)
12
  app = FastAPI()
13
 
14
  # ✅ Model name
15
- MODEL_NAME = "facebook/blenderbot-1B-distill"
16
 
17
  # Load tokenizer & model once
18
  tokenizer = BlenderbotTokenizer.from_pretrained(MODEL_NAME, cache_dir=CACHE_DIR)
 
12
  app = FastAPI()
13
 
14
  # ✅ Model name
15
+ MODEL_NAME = "facebook/blenderbot-400M-distill"
16
 
17
  # Load tokenizer & model once
18
  tokenizer = BlenderbotTokenizer.from_pretrained(MODEL_NAME, cache_dir=CACHE_DIR)