prithivMLmods commited on
Commit
5b01bf4
·
verified ·
1 Parent(s): d1a0e79

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -156,7 +156,7 @@ def process_history(history: list[dict]) -> list[dict]:
156
  current_user_content.append({"type": "image", "url": content[0]})
157
  return messages
158
 
159
- @spaces.GPU(duration=120)
160
  def run(message: dict, history: list[dict], system_prompt: str = "", max_new_tokens: int = 512) -> Iterator[str]:
161
  if not validate_media_constraints(message, history):
162
  yield ""
@@ -340,4 +340,4 @@ demo = gr.ChatInterface(
340
  )
341
 
342
  if __name__ == "__main__":
343
- demo.launch()
 
156
  current_user_content.append({"type": "image", "url": content[0]})
157
  return messages
158
 
159
+ @spaces.GPU(duration=90)
160
  def run(message: dict, history: list[dict], system_prompt: str = "", max_new_tokens: int = 512) -> Iterator[str]:
161
  if not validate_media_constraints(message, history):
162
  yield ""
 
340
  )
341
 
342
  if __name__ == "__main__":
343
+ demo.launch(debug=True)