Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -57,7 +57,7 @@ print(f"[ForgeCaptions] Gradio version: {gr.__version__}")
|
|
| 57 |
# - processor on CPU (safe)
|
| 58 |
# - model created lazily inside @gpu functions
|
| 59 |
# ------------------------------
|
| 60 |
-
processor = AutoProcessor.from_pretrained(MODEL_PATH
|
| 61 |
_MODEL = None
|
| 62 |
_DEVICE = "cpu"
|
| 63 |
_DTYPE = torch.float32
|
|
|
|
| 57 |
# - processor on CPU (safe)
|
| 58 |
# - model created lazily inside @gpu functions
|
| 59 |
# ------------------------------
|
| 60 |
+
processor = AutoProcessor.from_pretrained(MODEL_PATH)
|
| 61 |
_MODEL = None
|
| 62 |
_DEVICE = "cpu"
|
| 63 |
_DTYPE = torch.float32
|