Update app.py
Browse files
app.py
CHANGED
|
@@ -8,9 +8,6 @@ import numpy as np
|
|
| 8 |
from utils.goat import call_inference
|
| 9 |
import io
|
| 10 |
|
| 11 |
-
# Suppress warnings
|
| 12 |
-
warnings.filterwarnings("ignore", category=UserWarning, message="Using a slow image processor as `use_fast` is unset")
|
| 13 |
-
|
| 14 |
# Ensure using GPU if available
|
| 15 |
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
| 16 |
|
|
|
|
| 8 |
from utils.goat import call_inference
|
| 9 |
import io
|
| 10 |
|
|
|
|
|
|
|
|
|
|
| 11 |
# Ensure using GPU if available
|
| 12 |
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
| 13 |
|