KingNish commited on
Commit
0afabcf
1 Parent(s): cb3cac3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -36,7 +36,7 @@ from transformers import AutoProcessor
36
  model3 = AutoModel.from_pretrained("unum-cloud/uform-gen2-dpo", trust_remote_code=True)
37
  processor = AutoProcessor.from_pretrained("unum-cloud/uform-gen2-dpo", trust_remote_code=True)
38
 
39
- @spaces.GPU(enable_queue=False)
40
  def videochat(image3, prompt3):
41
  inputs = processor(text=[prompt3], images=[image3], return_tensors="pt")
42
  with torch.inference_mode():
@@ -304,7 +304,7 @@ def extract_images_from_msg_list(msg_list):
304
  return all_images
305
 
306
 
307
- @spaces.GPU(duration=30, enable_queue=False)
308
  def model_inference(
309
  user_prompt,
310
  chat_history,
 
36
  model3 = AutoModel.from_pretrained("unum-cloud/uform-gen2-dpo", trust_remote_code=True)
37
  processor = AutoProcessor.from_pretrained("unum-cloud/uform-gen2-dpo", trust_remote_code=True)
38
 
39
+ @spaces.GPU(queue=False)
40
  def videochat(image3, prompt3):
41
  inputs = processor(text=[prompt3], images=[image3], return_tensors="pt")
42
  with torch.inference_mode():
 
304
  return all_images
305
 
306
 
307
+ @spaces.GPU(duration=30, queue=False)
308
  def model_inference(
309
  user_prompt,
310
  chat_history,