VanYsa commited on
Commit
19df411
·
1 Parent(s): 0d3852f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -144,12 +144,12 @@ def bot(history,message):
144
  yield history
145
 
146
 
147
- @spaces.GPU(duration=120)
148
  def bot_response(message: str,
149
  history: list,
150
  temperature: float,
151
  max_new_tokens: int
152
- ) -> str: # type: ignore
153
  """
154
  Generate a streaming response using the llama3-8b model.
155
  Args:
 
144
  yield history
145
 
146
 
147
+ @spaces.GPU()
148
  def bot_response(message: str,
149
  history: list,
150
  temperature: float,
151
  max_new_tokens: int
152
+ ) -> str:
153
  """
154
  Generate a streaming response using the llama3-8b model.
155
  Args: