Update app.py
Browse files
app.py
CHANGED
@@ -43,8 +43,8 @@ dictionary = ['computer science', 'economics', 'Electrical Engineering and Syste
|
|
43 |
'statistics']
|
44 |
if filter(title, abstract):
|
45 |
model, tokenizer = Model()
|
46 |
-
|
47 |
-
|
48 |
st.markdown(f"{model}")
|
49 |
|
50 |
|
|
|
43 |
'statistics']
|
44 |
if filter(title, abstract):
|
45 |
model, tokenizer = Model()
|
46 |
+
logits = Predict(model, tokenizer, text)
|
47 |
+
Print(logits, dictionary)
|
48 |
st.markdown(f"{model}")
|
49 |
|
50 |
|