WesanCZE commited on
Commit
132c981
·
verified ·
1 Parent(s): 7a656c7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,8 +1,8 @@
1
  import gradio as gr
2
  from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
3
 
4
- # Použití otevřené verze Mistral 7B
5
- model_name = "open-mistral/open-mistral-7b"
6
  tokenizer = AutoTokenizer.from_pretrained(model_name)
7
  model = AutoModelForCausalLM.from_pretrained(model_name)
8
 
 
1
  import gradio as gr
2
  from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
3
 
4
+ # Použití TinyLlama 1.1B Chat (zcela zdarma)
5
+ model_name = "TinyLlama/TinyLlama-1.1B-Chat-v1.0"
6
  tokenizer = AutoTokenizer.from_pretrained(model_name)
7
  model = AutoModelForCausalLM.from_pretrained(model_name)
8