aaditya commited on
Commit
023be4e
·
1 Parent(s): 4f66887

Add application file

Browse files
Files changed (1) hide show
  1. app.py +9 -6
app.py CHANGED
@@ -31,17 +31,20 @@ class Data(Resource):
31
 
32
  class Datas(Resource):
33
  def post(self):
34
- parser.add_argument('dataset_name', type=str, required=True)
35
  args = parser.parse_args()
 
 
 
 
 
 
 
36
 
37
- print(args)
38
- dataset_name = args['dataset_name']
39
- DATASET_CONSTANT['data'].append(dataset_name)
40
- return {'status': 'Dataset added successfully'}
41
 
42
  api.add_resource(Status, '/')
43
  api.add_resource(Data, '/data')
44
- api.add_resource(Datas, '/datas')
45
 
46
 
47
 
 
31
 
32
  class Datas(Resource):
33
  def post(self):
34
+ parser.add_argument('prompt', type=str, required=True)
35
  args = parser.parse_args()
36
+ prompt_ = args['prompt']
37
+
38
+ try:
39
+ return_output = len(encoder.encode(prompt_))
40
+ return {'status': '1', 'token_legth': return_output}
41
+ except Exception as e:
42
+ return {'status': '-1'}
43
 
 
 
 
 
44
 
45
  api.add_resource(Status, '/')
46
  api.add_resource(Data, '/data')
47
+ api.add_resource(Datas, '/get_size')
48
 
49
 
50