Skip to content

Commit

Permalink
update
Browse files Browse the repository at this point in the history
Signed-off-by: Vivek jha <[email protected]>
  • Loading branch information
duckling69 committed May 19, 2024
1 parent c4ccb34 commit 9ba551a
Show file tree
Hide file tree
Showing 5 changed files with 10 additions and 18 deletions.
6 changes: 4 additions & 2 deletions app.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,12 +2,14 @@
import streamlit as st
import numpy as np
import tensorflow as tf
from tensorflow.keras.models import load_model
from keras.models import load_model
from tensorflow.keras.preprocessing.sequence import pad_sequences
import pickle
import os
print("Current working directory:", os.getcwd())

# Load the model, tokenizer, and label encoder
model = load_model('chatbot_model.h5')
model = load_model('/workspaces/OptimalYou-Chat/chatbot_model.h5')

with open('tokenizer.pickle', 'rb') as handle:
tokenizer = pickle.load(handle)
Expand Down
10 changes: 0 additions & 10 deletions application.py

This file was deleted.

2 changes: 1 addition & 1 deletion docker-compose.yml
Original file line number Diff line number Diff line change
Expand Up @@ -12,4 +12,4 @@ services:
if [ -f /workspace/requirements.txt ]; then pip install --user -r /workspace/requirements.txt; fi;
pip install --user streamlit;
export PATH=\$PATH:/root/.local/bin;
streamlit run /workspace/application.py --server.enableCORS false --server.enableXsrfProtection false"
streamlit run /workspace/app.py --server.enableCORS false --server.enableXsrfProtection false"
5 changes: 1 addition & 4 deletions req.txt
Original file line number Diff line number Diff line change
@@ -1,4 +1 @@
flask
google-cloud-vertex-ai
pandas
transformers

5 changes: 4 additions & 1 deletion requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -30,4 +30,7 @@ python-engineio==4.9.0
python-socketio==5.11.1
streamlit==1.33.0
toml==0.10.2
vertexai==1.49.0
vertexai==1.49.0
flask
transformers
tensorflow

0 comments on commit 9ba551a

Please sign in to comment.