Tonic commited on
Commit
e11a899
1 Parent(s): 3a69f52

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -1,4 +1,4 @@
1
- ```python
2
  import json
3
  import torch
4
  import gradio as gr
@@ -60,6 +60,7 @@ def build_prompt(task_instruction: str, format_instruction: str, tools: list, qu
60
  prompt += build_conversation_history_prompt(conversation_history)
61
  return prompt
62
 
 
63
  def generate_response(tools_input, query):
64
  try:
65
  tools = json.loads(tools_input)
 
1
+ import spaces
2
  import json
3
  import torch
4
  import gradio as gr
 
60
  prompt += build_conversation_history_prompt(conversation_history)
61
  return prompt
62
 
63
+ @spaces.GPU
64
  def generate_response(tools_input, query):
65
  try:
66
  tools = json.loads(tools_input)