nornorr commited on
Commit
4fd2e54
1 Parent(s): 4eba561

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -1,7 +1,7 @@
1
  import streamlit as st
2
  from transformers import pipeline
3
 
4
- classifier = pipeline("text-classification", model="DistilBERT base uncased finetuned SST-2")
5
  def main():
6
  st.title("Text-classification")
7
 
@@ -14,4 +14,6 @@ def main():
14
  st.json(results)
15
 
16
  if __name__ == "__main__":
17
- main()
 
 
 
1
  import streamlit as st
2
  from transformers import pipeline
3
 
4
+ classifier = pipeline("conversational", model="robinhad/gpt2-uk-conversational")
5
  def main():
6
  st.title("Text-classification")
7
 
 
14
  st.json(results)
15
 
16
  if __name__ == "__main__":
17
+ main()
18
+
19
+ """'audio-classification', 'automatic-speech-recognition', 'conversational', 'document-question-answering', 'feature-extraction', 'fill-mask', 'image-classification', 'image-segmentation', 'image-to-text', 'ner', 'object-detection', 'question-answering', 'sentiment-analysis', 'summarization', 'table-question-answering', 'text-classification', 'text-generation', 'text2text-generation', 'token-classification', 'translation', 'visual-question-answering', 'vqa', 'zero-shot-classification', 'zero-shot-image-classification', 'translation_XX_to_YY'"""