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