stinoco commited on
Commit
ad532b0
1 Parent(s): a87c588

Added classification models for subcategories

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,8 +1,8 @@
1
  import gradio as gr
2
  import numpy as np
3
  from transformers import pipeline
4
- from utils.tokenizer import tokenizer
5
- from utils.lstm import lstm
6
  from utils.load_model import load_model
7
  from utils.production_model import ProductionModel
8
 
 
1
  import gradio as gr
2
  import numpy as np
3
  from transformers import pipeline
4
+ from utils.tokenizer import Tokenizer
5
+ from utils.lstm import LSTM
6
  from utils.load_model import load_model
7
  from utils.production_model import ProductionModel
8