initial

app.py

@@ -0,0 +1,52 @@
+import streamlit as st
+from transformers import pipeline
+
+option = st.selectbox(
+    "Select an Option",
+    [
+        "Classify Text",
+        "Question Answering",
+        "Text Generation",
+        "Named Entity Recognition",
+        "Summarization",
+        "Translation",
+    ],
+)
+
+if option == "Classify Text":
+    text = st.text_area(label="Enter text")
+    if text:
+        classifier = pipeline("sentiment-analysis")
+        answer = classifier(text)
+        st.write(answer)
+elif option == "Question Answering":
+    q_a = pipeline("question-answering")
+    context = st.text_area(label="Enter context")
+    question = st.text_area(label="Enter question")
+    if context and question:
+        answer = q_a({"question": question, "context": context})
+        st.write(answer)
+elif option == "Text Generation":
+    text = st.text_area(label="Enter text")
+    if text:
+        text_generator = pipeline("text-generation")
+        answer = text_generator(text)
+        st.write(answer)
+elif option == "Named Entity Recognition":
+    text = st.text_area(label="Enter text")
+    if text:
+        ner = pipeline("ner")
+        answer = ner(text)
+        st.write(answer)
+elif option == "Summarization":
+    summarizer = pipeline("summarization")
+    article = st.text_area(label="Paste Article")
+    if article:
+        summary = summarizer(article, max_length=400, min_length=30)
+        st.write(summary)
+elif option == "Translation":
+    translator = pipeline("translation_en_to_de")
+    text = st.text_area(label="Enter text")
+    if text:
+        translation = translator(text)
+        st.write(translation)

polyaxonfiles/app.yaml

@@ -0,0 +1,12 @@
+version: 1.1
+kind: component
+name: plx-streamlit-huggingface-app
+tags: [streamlit, huggingface]
+run:
+  kind: service
+  ports: [8501]
+  rewritePath: true
+  container:
+    image: avivkri/polyaxon-examples:plx-streamlit-huggingface-app
+    command: [streamlit, run, app.py]
+

polyaxonfiles/build.yaml

@@ -0,0 +1,15 @@
+version: 1.1
+kind: operation
+name: build-plx-streamlit-huggingface-app
+params:
+  destination:
+    connection: "docker-connection"
+    value: polyaxon-examples:plx-streamlit-huggingface-app
+  context:
+    value: "{{ globals.artifacts_path }}/plx-streamlit-huggingface-app"
+runPatch:
+  init:
+    - git:
+        url: "https://github.com/avivkri/plx-streamlit-huggingface-app"
+hubRef: kaniko
+

requirements.txt

@@ -0,0 +1,3 @@
+streamlit==0.70.0
+transformers==3.5.1
+typing-extensions==3.7.4.3