Subhajit Chakraborty commited on
Commit
0ee2512
·
1 Parent(s): 0e600e7
Files changed (1) hide show
  1. app.py +2 -2
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("microsoft/phi-2")
8
- model = AutoModelForCausalLM.from_pretrained("microsoft/phi-2")
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()