Hugging Face's logo

Spaces:
Bob-Potato
/
MetaGPT_AI
Runtime error

App Files Community
Bob-Potato commited on
Commit
6e899e3
·
verified ·
1 Parent(s): 433c9c6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ app = FastAPI()
10
  # ---------------------------
11
  # 1. Încarcă modelul Gemma 1B
12
  # ---------------------------
13
- MODEL_NAME = "google/gemma-3-1b-it" # modelul mic Gemma 1B
14
  tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
15
  model = AutoModelForCausalLM.from_pretrained(MODEL_NAME)
16
 
 
10
  # ---------------------------
11
  # 1. Încarcă modelul Gemma 1B
12
  # ---------------------------
13
+ MODEL_NAME = "distilgpt2" # modelul mic Gemma 1B
14
  tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
15
  model = AutoModelForCausalLM.from_pretrained(MODEL_NAME)
16