Wonder-Griffin commited on
Commit
51599c2
1 Parent(s): 4d39168

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -4,8 +4,8 @@ from huggingface_hub import InferenceClient
4
  # Load model directly
5
  from transformers import AutoTokenizer, AutoModelForCausalLM
6
 
7
- tokenizer = AutoTokenizer.from_pretrained("Wonder-Griffin/Shorsey-T2000")
8
- model = AutoModel.from_pretrained("Wonder-Griffin/Shorsey-T2000")
9
 
10
 
11
  def respond(
 
4
  # Load model directly
5
  from transformers import AutoTokenizer, AutoModelForCausalLM
6
 
7
+ tokenizer = AutoTokenizer.from_pretrained("Wonder-Griffin/TraXLMistral")
8
+ model = AutoModelForCausalLM.from_pretrained("Wonder-Griffin/TraXLMistral")
9
 
10
 
11
  def respond(