Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
4ed3940
1
Parent(s):
5193c2b
Add application file
Browse files
app.py
CHANGED
@@ -13,9 +13,9 @@ model = Qwen2_5_VLForConditionalGeneration.from_pretrained(
|
|
13 |
|
14 |
processor = AutoProcessor.from_pretrained(local_path)
|
15 |
|
|
|
16 |
# Function to process image and text and generate the output
|
17 |
@torch.inference_mode()
|
18 |
-
@spaces.GPU(duration=120) # Specify a duration to avoid timeout
|
19 |
def generate_output(image, text, button_click):
|
20 |
# Prepare input data
|
21 |
messages = [
|
|
|
13 |
|
14 |
processor = AutoProcessor.from_pretrained(local_path)
|
15 |
|
16 |
+
print("load successfully")
|
17 |
# Function to process image and text and generate the output
|
18 |
@torch.inference_mode()
|
|
|
19 |
def generate_output(image, text, button_click):
|
20 |
# Prepare input data
|
21 |
messages = [
|