mssaidat commited on
Commit
47c889a
·
verified ·
1 Parent(s): 07e9fc6

Update app.py

Browse files

Changed the Model Name

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -1,7 +1,7 @@
1
  import gradio as gr
2
  from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
3
 
4
- MODEL_NAME = "openai/gpt-oss-20b"
5
 
6
  # Load tokenizer & model
7
  #tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
 
1
  import gradio as gr
2
  from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
3
 
4
+ MODEL_NAME = "TinyLlama/TinyLlama-1.1B-Chat-v1.0"
5
 
6
  # Load tokenizer & model
7
  #tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)