Skip to content

Commit

Permalink
updated main.py Dockerfile requirements.txt
Browse files Browse the repository at this point in the history
  • Loading branch information
arihant1805 committed Jun 15, 2024
1 parent 6898867 commit 5b71333
Showing 1 changed file with 6 additions and 0 deletions.
6 changes: 6 additions & 0 deletions app/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,6 +10,12 @@
model=tensorflow.keras.models.load_model('model.h5')
tokenizer = pickle.load(open('Tokenizer.pkl', 'rb'))

@App.get('/')
def default():
return {
'msg' : 'The App started'
}

@App.post('/')
def spamDetector(msg: str):
inp = np.array([msg])
Expand Down

0 comments on commit 5b71333

Please sign in to comment.