laurentiubp commited on
Commit
a9c863c
·
verified ·
1 Parent(s): 9bea0be

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -1,3 +1,4 @@
 
1
  import gradio as gr
2
  from huggingface_hub import InferenceClient
3
 
@@ -7,6 +8,7 @@ For more information on `huggingface_hub` Inference API support, please check th
7
  client = InferenceClient("catallama/CataLlama-v0.1-Instruct-DPO")
8
 
9
 
 
10
  def respond(
11
  message,
12
  history: list[tuple[str, str]],
 
1
+ import spaces
2
  import gradio as gr
3
  from huggingface_hub import InferenceClient
4
 
 
8
  client = InferenceClient("catallama/CataLlama-v0.1-Instruct-DPO")
9
 
10
 
11
+ @spaces.GPU(duration=120)
12
  def respond(
13
  message,
14
  history: list[tuple[str, str]],