Ezi Ozoani commited on
Commit
1f1008b
·
1 Parent(s): ae1ed3c

updating requirements

Browse files
Files changed (1) hide show
  1. models.py +28 -0
models.py ADDED
@@ -0,0 +1,28 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import os
2
+ from pydantic import BaseModel
3
+ from typing import Literal
4
+
5
+ class Config(BaseModel):
6
+ target_namespace: str
7
+ input_dataset: str
8
+ input_model: str
9
+ autotrain_project_prefix: str
10
+
11
+
12
+ class WebhookPayloadEvent(BaseModel):
13
+ action: Literal["create", "update", "delete"]
14
+ scope: str
15
+
16
+ class WebhookPayloadRepo(BaseModel):
17
+ type: Literal["dataset", "model", "space"]
18
+ name: str
19
+ id: str
20
+ private: bool
21
+ headSha: str
22
+
23
+ class WebhookPayload(BaseModel):
24
+ event: WebhookPayloadEvent
25
+ repo: WebhookPayloadRepo
26
+
27
+
28
+ config = Config.parse_file(os.path.join(os.getcwd(), "config.json"))