Gyr0MAN commited on
Commit
0f5fcd6
β€’
1 Parent(s): 1a6ef81

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -1,9 +1,10 @@
1
  import gradio as gr
2
  import torch
3
  import spaces
4
- from transformers import AutoTokenizer, AutoModelForCausalLM
 
5
 
6
- tokenizer = AutoTokenizer.from_pretrained("ping98k/typhoon-7b-rag-instruct-th")
7
  model = AutoModelForCausalLM.from_pretrained("ping98k/typhoon-7b-rag-instruct-th")
8
 
9
  @spaces.GPU(duration=120)
 
1
  import gradio as gr
2
  import torch
3
  import spaces
4
+ from tokenizers import Tokenizer
5
+ from transformers import AutoModelForCausalLM
6
 
7
+ tokenizer = Tokenizer.from_pretrained("ping98k/typhoon-7b-rag-instruct-th")
8
  model = AutoModelForCausalLM.from_pretrained("ping98k/typhoon-7b-rag-instruct-th")
9
 
10
  @spaces.GPU(duration=120)