Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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(
|
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,
|
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,
|