VictorSanh commited on
Commit
5e5d5d1
1 Parent(s): 792e854

Update app_dialogue.py

Browse files
Files changed (1) hide show
  1. app_dialogue.py +2 -0
app_dialogue.py CHANGED
@@ -2,6 +2,7 @@ import copy
2
  import hashlib
3
  import os
4
  import re
 
5
  import subprocess
6
  import torch
7
  import PIL
@@ -427,6 +428,7 @@ with gr.Blocks(title="IDEFICS Playground", theme=gr.themes.Base()) as demo:
427
  outputs=top_p,
428
  )
429
 
 
430
  def model_inference(
431
  model_selector,
432
  user_prompt_str,
 
2
  import hashlib
3
  import os
4
  import re
5
+ import spaces
6
  import subprocess
7
  import torch
8
  import PIL
 
428
  outputs=top_p,
429
  )
430
 
431
+ @spaces.GPU(duration=180)
432
  def model_inference(
433
  model_selector,
434
  user_prompt_str,