Sambhavnoobcoder commited on
Commit
625452f
·
1 Parent(s): 132692f

tf.keras.models.load_model added

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -1,8 +1,9 @@
1
  import gradio as gr
 
2
  from fastai.vision.all import *
3
 
4
  # Load the model
5
- learn = load_learner('sentimentality.h5')
6
 
7
  # Prediction function
8
  def make_prediction(user_sentence):
 
1
  import gradio as gr
2
+ import tensorflow as tf
3
  from fastai.vision.all import *
4
 
5
  # Load the model
6
+ learn = tf.keras.models.load_model('sentimentality.h5')
7
 
8
  # Prediction function
9
  def make_prediction(user_sentence):