uptaleend commited on
Commit
c748b09
1 Parent(s): 46a138c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,8 +1,8 @@
1
  from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
2
  from threading import Thread
3
 
4
- model = AutoModelForCausalLM.from_pretrained("Qwen/Qwen2-0.5B-Instruct")
5
- tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen2-0.5B-Instruct")
6
  streamer = TextIteratorStreamer(tokenizer, skip_prompt=True)
7
 
8
  import re
 
1
  from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
2
  from threading import Thread
3
 
4
+ model = AutoModelForCausalLM.from_pretrained("ystemsrx/Qwen2-Boundless")
5
+ tokenizer = AutoTokenizer.from_pretrained("ystemsrx/Qwen2-Boundless")
6
  streamer = TextIteratorStreamer(tokenizer, skip_prompt=True)
7
 
8
  import re