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():
|