Hugging Face's logo

Spaces:
BeveledCube
/
bevelapi
Runtime error

App Files Community
BeveledCube commited on
Commit
bc27fb1
·
verified ·
1 Parent(s): 90151e1

Increased model size to

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -5,7 +5,7 @@ from fastapi import FastAPI
5
 
6
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
7
 
8
- model_name = "facebook/blenderbot-400M-distill"
9
 
10
  # https://huggingface.co/models?sort=trending&search=facebook%2Fblenderbo
11
  # facebook/blenderbot-3B
 
5
 
6
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
7
 
8
+ model_name = "facebook/blenderbot-1B-distill"
9
 
10
  # https://huggingface.co/models?sort=trending&search=facebook%2Fblenderbo
11
  # facebook/blenderbot-3B