momergul commited on
Commit
33f8437
1 Parent(s): df7ebab
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -36,7 +36,7 @@ def initialize_game() -> List[List[str]]:
36
 
37
  return list(zip(speaker_images, listener_images, targets, roles))
38
 
39
- @spaces.GPU
40
  def get_model_response(
41
  model, adapter_name, processor, index_to_token, role: str,
42
  image_paths: List[str], user_message: str = "", target_image: str = ""
 
36
 
37
  return list(zip(speaker_images, listener_images, targets, roles))
38
 
39
+ @spaces.GPU(duration=120)
40
  def get_model_response(
41
  model, adapter_name, processor, index_to_token, role: str,
42
  image_paths: List[str], user_message: str = "", target_image: str = ""