Upload app.py

app.py

@@ -39,6 +39,7 @@ orange = [197/255, 90/255, 17/255]
 # Pipelines definieren
 en_de_translator = pipeline("translation_de_to_en", model='google/bert2bert_L-24_wmt_de_en')
 qa_pipeline = pipeline("question-answering", model='deepset/gelectra-base-germanquad')
+sentiment = pipeline("text-classification", model='oliverguhr/german-sentiment-bert')
 
 tab1, tab2, tab3, tab4 = st.tabs(
     ["Künstliche Neuronale Netze", "Wortvektoren Stimmung", "Wörter Maskieren", "Demos"])
@@ -367,7 +368,12 @@ with tab4:
             a5 = en_de_translator(string_3)
             st.text(a5)
 
+    #########################################################
     st.text('')
+    st.markdown("""<hr style="height:10px;border:none;color:#333;background-color:#333;" /> """,
+                unsafe_allow_html=True)
+    st.text('')
+
     text_input_4 = 'Was ist der Schwerpunkt?'
 
     string_4 = st.text_area('Frage zum Kontext beantworten', value=text_input_4, height=25)
@@ -381,3 +387,30 @@ with tab4:
 
             a5 = qa_pipeline(question=string_4, context=string_5)
             st.text(a5)
+
+    ############################################################
+    st.text('')
+    st.markdown("""<hr style="height:10px;border:none;color:#333;background-color:#333;" /> """,
+                unsafe_allow_html=True)
+    st.text('')
+
+    text_input_7 = 'Wir lieben Data Science.?'
+
+    string_7 = st.text_area('Stimmungsanalyse', value=text_input_7, height=25)
+
+    if st.button('Ein fertig trainiertes Transformer-Modell von HuggingFace anwenden', key=4):
+        with st.spinner('Die Beurteilung der Stimmung kann einige Sekunden dauern ...'):
+
+            a5 = sentiment(string_7)
+            print(a5[0]['label'])
+
+    st.text('')
+    st.text('Verwendete Modelle:')
+    st.text('\nÜbersetzung:')
+    st.text('google/bert2bert_L-24_wmt_de_en, Authors: Sascha Rothe, Shashi Narayan, Aliaksei Severyn')
+
+    st.text('\nFrage beantworten:')
+    st.text('deepset/gelectra-base-germanquad, Authors: Timo Möller, Julian Risch, Malte Pietsch')
+
+    st.text('\nStimmung:')
+    st.text('oliverguhr/german-sentiment-bert, Authors:  Oliver Guhr, Anne-Kathrin Schumann, Frank Bahrmann, Hans Joachim Böhme')