Vikrantyadav11234 commited on
Commit
c1202d1
·
verified ·
1 Parent(s): 14ed9d4

Delete files tokenizer.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. tokenizer.py +0 -23
tokenizer.py DELETED
@@ -1,23 +0,0 @@
1
- from flask import Flask, request, jsonify
2
- from transformers import AutoTokenizer
3
-
4
- app = Flask(__name__)
5
-
6
- # Load tokenizer
7
- tokenizer = AutoTokenizer.from_pretrained("verbalyze/Text2Text_Conversation_Pretrained_V2__model")
8
-
9
- @app.route('/tokenize', methods=['POST'])
10
- def tokenize():
11
- data = request.json
12
- prompt = data.get("prompt", "")
13
-
14
- if not prompt:
15
- return jsonify({"error": "No prompt provided"}), 400
16
-
17
- # Tokenize the prompt
18
- input_ids = tokenizer(prompt, return_tensors='pt').input_ids.tolist()
19
-
20
- return jsonify({"input_ids": input_ids})
21
-
22
- if __name__ == '__main__':
23
- app.run(host='0.0.0.0', port=5001)