Hugging Face's logo

Spaces:
NoesisLab
/
Kai-30B-Instruct
Running on Zero

App Files Community
OzTianlu commited on
Commit
1064c57
·
verified ·
1 Parent(s): ba7f9ae

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -3,11 +3,11 @@ import torch
3
  from threading import Thread
4
  from transformers import AutoTokenizer, AutoModelForCausalLM, TextIteratorStreamer
5
  import gradio as gr
6
- MODEL_ID = "NoesisLab/Kai-3B-Instruct"
7
 
8
  tokenizer = AutoTokenizer.from_pretrained(MODEL_ID, trust_remote_code=True)
9
  model = AutoModelForCausalLM.from_pretrained(
10
- "NoesisLab/Kai-3B-Instruct",
11
  )
12
 
13
  @spaces.GPU
@@ -56,8 +56,8 @@ def respond(message, history):
56
 
57
  demo = gr.ChatInterface(
58
  fn=respond,
59
- title="Chat with Kai-3B-Instruct",
60
- description="Chat with NoesisLab/Kai-3B-Instruct",
61
  )
62
 
63
  if __name__ == "__main__":
 
3
  from threading import Thread
4
  from transformers import AutoTokenizer, AutoModelForCausalLM, TextIteratorStreamer
5
  import gradio as gr
6
+ MODEL_ID = "NoesisLab/Kai-30B-Instruct"
7
 
8
  tokenizer = AutoTokenizer.from_pretrained(MODEL_ID, trust_remote_code=True)
9
  model = AutoModelForCausalLM.from_pretrained(
10
+ "NoesisLab/Kai-30B-Instruct",
11
  )
12
 
13
  @spaces.GPU
 
56
 
57
  demo = gr.ChatInterface(
58
  fn=respond,
59
+ title="Chat with Kai-30B-Instruct",
60
+ description="Chat with NoesisLab/Kai-30B-Instruct",
61
  )
62
 
63
  if __name__ == "__main__":