Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -8,6 +8,10 @@ from uuid import uuid4
|
|
8 |
|
9 |
app = Flask(__name__)
|
10 |
|
|
|
|
|
|
|
|
|
11 |
# Load the model once at startup
|
12 |
model_id = "Wan-AI/Wan2.1-T2V-1.3B-Diffusers"
|
13 |
vae = AutoencoderKLWan.from_pretrained(model_id, subfolder="vae", torch_dtype=torch.float32)
|
|
|
8 |
|
9 |
app = Flask(__name__)
|
10 |
|
11 |
+
@app.route('/')
|
12 |
+
def index():
|
13 |
+
return jsonify({"message": "Welcome to the Wan2.1 Video Generation API!", "status": "running"})
|
14 |
+
|
15 |
# Load the model once at startup
|
16 |
model_id = "Wan-AI/Wan2.1-T2V-1.3B-Diffusers"
|
17 |
vae = AutoencoderKLWan.from_pretrained(model_id, subfolder="vae", torch_dtype=torch.float32)
|