Spaces:
Build error
Build error
Commit ·
125da18
1
Parent(s): 7e6e555
Update app.py
Browse files
app.py
CHANGED
|
@@ -64,7 +64,7 @@ def clean_text(text):
|
|
| 64 |
|
| 65 |
|
| 66 |
def predict(text):
|
| 67 |
-
model = keras.models.load_model('
|
| 68 |
pred = model.predict(text)
|
| 69 |
pred_class = np.argmax(pred, axis=-1)
|
| 70 |
return labels.get(pred_class[0])
|
|
|
|
| 64 |
|
| 65 |
|
| 66 |
def predict(text):
|
| 67 |
+
model = keras.models.load_model('model/cnn_model.h5')
|
| 68 |
pred = model.predict(text)
|
| 69 |
pred_class = np.argmax(pred, axis=-1)
|
| 70 |
return labels.get(pred_class[0])
|