Spaces:
Sleeping
Sleeping
Subhajit Chakraborty
commited on
Commit
·
0e600e7
1
Parent(s):
678a4fc
- app.py +2 -1
- requirements.txt +1 -0
app.py
CHANGED
|
@@ -1,8 +1,9 @@
|
|
| 1 |
from flask import Flask, request, jsonify
|
| 2 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
|
|
|
| 3 |
|
| 4 |
app = Flask(__name__)
|
| 5 |
-
|
| 6 |
tokenizer = AutoTokenizer.from_pretrained("microsoft/phi-2")
|
| 7 |
model = AutoModelForCausalLM.from_pretrained("microsoft/phi-2")
|
| 8 |
@app.route('/chat', methods=['POST'])
|
|
|
|
| 1 |
from flask import Flask, request, jsonify
|
| 2 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 3 |
+
from flask_cors import CORS
|
| 4 |
|
| 5 |
app = Flask(__name__)
|
| 6 |
+
CORS(app)
|
| 7 |
tokenizer = AutoTokenizer.from_pretrained("microsoft/phi-2")
|
| 8 |
model = AutoModelForCausalLM.from_pretrained("microsoft/phi-2")
|
| 9 |
@app.route('/chat', methods=['POST'])
|
requirements.txt
CHANGED
|
@@ -1,3 +1,4 @@
|
|
| 1 |
flask
|
|
|
|
| 2 |
transformers
|
| 3 |
torch
|
|
|
|
| 1 |
flask
|
| 2 |
+
flask-cors
|
| 3 |
transformers
|
| 4 |
torch
|