Initial Draft

app.py

@@ -47,7 +47,7 @@ def sentiment_analysis():
 
 def sentiment_analysis_btn():
     classifier = pipeline("sentiment-analysis")
-    st.session_state.output_text=classifier(st.session_state.input_text)
+    st.session_state.output_text=str(classifier(st.session_state.input_text))
 
 def zero_shot_classification():
     st.session_state.input_text="This is a course about the transformers library"
@@ -55,10 +55,10 @@ def zero_shot_classification():
 
 def zero_shot_classification_btn():
     classifier = pipeline("zero-shot-classification")
-    st.session_state.output_text=classifier(
+    st.session_state.output_text=str(classifier(
         st.session_state.input_text,
         candidate_labels=["education", "politics", "business"],
-    )
+    ))
 
 def text_generation():
     st.session_state.input_text="In this course, we will teach you how to"
@@ -66,7 +66,7 @@ def text_generation():
 
 def text_generation_btn():
     generator = pipeline("text-generation")
-    st.session_state.output_text=generator(st.session_state.input_text)
+    st.session_state.output_text=str(generator(st.session_state.input_text))
 
 def mask_filling():
     st.session_state.input_text="This course will teach you all about <mask> models."
@@ -74,7 +74,7 @@ def mask_filling():
 
 def mask_filling_btn():
     unmasker = pipeline("fill-mask")
-    st.session_state.output_text=unmasker(st.session_state.input_text, top_k=2)
+    st.session_state.output_text=str(unmasker(st.session_state.input_text, top_k=2))
 
 def named_entity_recognition():
     st.session_state.input_text="My name is Sylvain and I work at Hugging Face in Brooklyn."
@@ -82,7 +82,7 @@ def named_entity_recognition():
 
 def named_entity_recognition_btn():
     ner = pipeline("ner", grouped_entities=True)
-    st.session_state.output_text=ner(st.session_state.input_text)
+    st.session_state.output_text=str(ner(st.session_state.input_text))
 
 def qna():
     st.session_state.input_text="My name is Sylvain and I work at Hugging Face in Brooklyn"
@@ -91,10 +91,10 @@ def qna():
 def qna_btn():
     question_answerer = pipeline("question-answering")
     st.session_state.output_text="Questionn: Where do I work?\n Answer: "
-    st.session_state.output_text=st.session_state.output_text + question_answerer(
+    st.session_state.output_text=st.session_state.output_text + str(question_answerer(
         question=st.session_state.input_text,
         context="My name is Sylvain and I work at Hugging Face in Brooklyn",
-    )
+    ))
 
 def summarization():
     st.session_state.input_text= """America has changed dramatically during recent years. Not only has the number of 
@@ -120,7 +120,7 @@ def summarization():
 
 def summarization_btn():
     summarizer = pipeline("summarization")
-    st.session_state.output_text=summarizer(st.session_state.input_text)
+    st.session_state.output_text=str(summarizer(st.session_state.input_text))
 
 def translation_eng2frn():
     st.session_state.input_text= "This course is produced by Hugging Face."
@@ -128,7 +128,7 @@ def translation_eng2frn():
 
 def translation_eng2frn_btn():
     translator = pipeline("translation", model="Helsinki-NLP/opus-mt-en-fr")
-    st.session_state.output_text=translator(st.session_state.input_text)
+    st.session_state.output_text=str(translator(st.session_state.input_text))
     
 ################################
 ####### Display of data ########