Spaces:
Sleeping
Sleeping
edbeeching
commited on
Commit
·
5d89dcd
1
Parent(s):
60d17e2
fix num samples in validation
Browse files
app.py
CHANGED
|
@@ -53,7 +53,7 @@ class GenerationRequest:
|
|
| 53 |
private: bool = False
|
| 54 |
num_retries: int = 0
|
| 55 |
|
| 56 |
-
def validate_request(request: GenerationRequest):
|
| 57 |
# checks that the request is valid
|
| 58 |
# - input dataset exists and can be accessed with the provided token
|
| 59 |
try:
|
|
@@ -99,6 +99,8 @@ def validate_request(request: GenerationRequest):
|
|
| 99 |
# check valid email address TODO: use py3-validate-email https://stackoverflow.com/questions/8022530/how-to-check-for-valid-email-address
|
| 100 |
if "@" not in request.email or "." not in request.email.split("@")[-1]:
|
| 101 |
raise Exception("Invalid email address")
|
|
|
|
|
|
|
| 102 |
|
| 103 |
|
| 104 |
def add_request_to_db(request: GenerationRequest):
|
|
@@ -248,7 +250,7 @@ def create_gradio_interface():
|
|
| 248 |
)
|
| 249 |
|
| 250 |
# check the input dataset exists and can be accessed with the provided token
|
| 251 |
-
validate_request(request)
|
| 252 |
add_request_to_db(request)
|
| 253 |
|
| 254 |
return "Request submitted successfully!"
|
|
|
|
| 53 |
private: bool = False
|
| 54 |
num_retries: int = 0
|
| 55 |
|
| 56 |
+
def validate_request(request: GenerationRequest) -> GenerationRequest:
|
| 57 |
# checks that the request is valid
|
| 58 |
# - input dataset exists and can be accessed with the provided token
|
| 59 |
try:
|
|
|
|
| 99 |
# check valid email address TODO: use py3-validate-email https://stackoverflow.com/questions/8022530/how-to-check-for-valid-email-address
|
| 100 |
if "@" not in request.email or "." not in request.email.split("@")[-1]:
|
| 101 |
raise Exception("Invalid email address")
|
| 102 |
+
|
| 103 |
+
return request
|
| 104 |
|
| 105 |
|
| 106 |
def add_request_to_db(request: GenerationRequest):
|
|
|
|
| 250 |
)
|
| 251 |
|
| 252 |
# check the input dataset exists and can be accessed with the provided token
|
| 253 |
+
request = validate_request(request)
|
| 254 |
add_request_to_db(request)
|
| 255 |
|
| 256 |
return "Request submitted successfully!"
|