kouki321 commited on
Commit
39af176
·
verified ·
1 Parent(s): 9b05251

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -2,7 +2,8 @@ import os
2
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
3
 
4
  # Model identifier
5
- model_id = "google/flan-t5-small"
 
6
  #"unsloth/mistral-7b-v0.2-bnb-4bit"
7
  #deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B
8
 
 
2
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
3
 
4
  # Model identifier
5
+ model_id = "sshleifer/tiny-gpt2"
6
+ #"google/flan-t5-small"
7
  #"unsloth/mistral-7b-v0.2-bnb-4bit"
8
  #deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B
9