cyrus28214 commited on
Commit
d342d8b
·
1 Parent(s): 48f8af8
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -1,11 +1,13 @@
1
  import gradio as gr
2
  from huggingface_hub import InferenceClient
 
3
 
4
  """
5
  For more information on `huggingface_hub` Inference API support, please check the docs: https://huggingface.co/docs/huggingface_hub/v0.22.2/en/guides/inference
6
  """
7
- client = InferenceClient("csfufu/Revisual-R1-final")
8
 
 
9
  def respond(
10
  message,
11
  history: list[tuple[str, str]],
@@ -16,7 +18,7 @@ def respond(
16
  ):
17
  messages = [{"role": "system", "content": system_message}]
18
 
19
- messages.extend(chat_history)
20
 
21
  messages.append({"role": "user", "content": message})
22
 
 
1
  import gradio as gr
2
  from huggingface_hub import InferenceClient
3
+ import spaces
4
 
5
  """
6
  For more information on `huggingface_hub` Inference API support, please check the docs: https://huggingface.co/docs/huggingface_hub/v0.22.2/en/guides/inference
7
  """
8
+ client = InferenceClient("HuggingFaceTB/SmolVLM-Instruct")
9
 
10
+ @spaces.GPU
11
  def respond(
12
  message,
13
  history: list[tuple[str, str]],
 
18
  ):
19
  messages = [{"role": "system", "content": system_message}]
20
 
21
+ messages.extend(history)
22
 
23
  messages.append({"role": "user", "content": message})
24