zhiqiulin commited on
Commit
0f8e3cc
·
verified ·
1 Parent(s): 7daba49

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -17,6 +17,7 @@ def generate(model_name, image, text):
17
  # Wrap the model call in a try-except block to capture and debug CUDA errors
18
  try:
19
  result = model_pipe(images=[image], texts=[text]).cpu() # Perform the model inference
 
20
  except RuntimeError as e:
21
  print(f"RuntimeError during model inference: {e}")
22
  raise e
 
17
  # Wrap the model call in a try-except block to capture and debug CUDA errors
18
  try:
19
  result = model_pipe(images=[image], texts=[text]).cpu() # Perform the model inference
20
+ print("Result", result)
21
  except RuntimeError as e:
22
  print(f"RuntimeError during model inference: {e}")
23
  raise e