Spaces:
Runtime error
Runtime error
Matt C
commited on
Commit
·
9db9938
1
Parent(s):
7cf8aab
tweak
Browse files
app.py
CHANGED
|
@@ -15,7 +15,8 @@ batch = tokenizer.encode('txt', return_tensors='pt')
|
|
| 15 |
|
| 16 |
# inference
|
| 17 |
result = model(batch)
|
| 18 |
-
print(result)
|
|
|
|
| 19 |
|
| 20 |
#fig = px.bar(result, x="", y="", orientation='h')
|
| 21 |
#fig.show()
|
|
|
|
| 15 |
|
| 16 |
# inference
|
| 17 |
result = model(batch)
|
| 18 |
+
print([d["text"] for d in result])
|
| 19 |
+
["Toxic", "Non-toxic"]
|
| 20 |
|
| 21 |
#fig = px.bar(result, x="", y="", orientation='h')
|
| 22 |
#fig.show()
|