snaramirez872 commited on
Commit
05235d4
·
1 Parent(s): fd05f24

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +25 -0
app.py ADDED
@@ -0,0 +1,25 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import streamlit as st
2
+ from transformers import pipeline
3
+ from transformers import AutoTokenizer as AT, AutoModelForSequenceClassification as AFSC
4
+
5
+ modName = "madhurjindal/autonlp-Gibberish-Detector-492513457" # Gibberish Detection Model from HuggingFace
6
+
7
+ mod = AFSC.from_pretrained(modName)
8
+ TKR = AT.from_pretrained(modName)
9
+
10
+ st.title("Gibberish Detector")
11
+
12
+ user_input = str(st.text_input("Enter some words", "pasghetti"))
13
+
14
+ st.markdown("Input was: ", str(user_input))
15
+
16
+ classifier = pipeline("sentiment-analysis", model=mod, tokenizer=TKR)
17
+ # result = classifier(["This is a sample text made by Sean Ramirez.", "This is another sample text."]) # leftover from initial testing
18
+
19
+ if user_input is not None:
20
+ col = st.columns(1)
21
+ predicts = pipeline("sentiment-analysis", model=mod, tokenizer=TKR)
22
+
23
+ col.header("Probabilities")
24
+ for p in predicts:
25
+ col.subheader(f"{ p['label']}: { round(p['score'] * 100, 1)}%")