Update app.py
Browse files
app.py
CHANGED
@@ -1,8 +1,9 @@
|
|
1 |
# app.py
|
2 |
-
from flask import Flask,
|
3 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
4 |
import torch
|
5 |
from functools import lru_cache
|
|
|
6 |
|
7 |
app = Flask(__name__)
|
8 |
|
@@ -35,7 +36,7 @@ def generate_response(prompt, model, tokenizer):
|
|
35 |
|
36 |
@app.route('/')
|
37 |
def home():
|
38 |
-
return
|
39 |
|
40 |
@app.route('/message', methods=['POST'])
|
41 |
def message():
|
|
|
1 |
# app.py
|
2 |
+
from flask import Flask, send_file, request, jsonify
|
3 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
4 |
import torch
|
5 |
from functools import lru_cache
|
6 |
+
import os
|
7 |
|
8 |
app = Flask(__name__)
|
9 |
|
|
|
36 |
|
37 |
@app.route('/')
|
38 |
def home():
|
39 |
+
return send_file('index.html')
|
40 |
|
41 |
@app.route('/message', methods=['POST'])
|
42 |
def message():
|