Fancy-MLLM commited on
Commit
4ed3940
·
1 Parent(s): 5193c2b

Add application file

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 = [