Ankit8544 commited on
Commit
87ff327
·
verified ·
1 Parent(s): 89fb97f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
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)