joermd commited on
Commit
5aef73d
·
verified ·
1 Parent(s): 8f76c97

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -1,8 +1,9 @@
1
  # app.py
2
- from flask import Flask, render_template, request, jsonify
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 render_template('index.html')
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():