Spaces:
Sleeping
Sleeping
Subhajit Chakraborty
commited on
Commit
·
0ee2512
1
Parent(s):
0e600e7
app.py
CHANGED
|
@@ -4,8 +4,8 @@ from flask_cors import CORS
|
|
| 4 |
|
| 5 |
app = Flask(__name__)
|
| 6 |
CORS(app)
|
| 7 |
-
tokenizer = AutoTokenizer.from_pretrained("
|
| 8 |
-
model = AutoModelForCausalLM.from_pretrained("
|
| 9 |
@app.route('/chat', methods=['POST'])
|
| 10 |
def chat():
|
| 11 |
data = request.get_json()
|
|
|
|
| 4 |
|
| 5 |
app = Flask(__name__)
|
| 6 |
CORS(app)
|
| 7 |
+
tokenizer = AutoTokenizer.from_pretrained("google/flan-t5-small")
|
| 8 |
+
model = AutoModelForCausalLM.from_pretrained("google/flan-t5-small")
|
| 9 |
@app.route('/chat', methods=['POST'])
|
| 10 |
def chat():
|
| 11 |
data = request.get_json()
|