tryout: switch to another pipline

.gitignore

@@ -0,0 +1 @@
+env

README.md

@@ -8,5 +8,7 @@ sdk_version: 1.39.0
 app_file: app.py
 pinned: false
 ---
-
+## how to locally run to debug
+1. pip install requirements.txt
+2. streamlit run app.py 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -11,19 +11,16 @@ tokenizer = AutoTokenizer.from_pretrained(base_model_name)
 model = BartForConditionalGeneration.from_pretrained(model_name)
 
 # Streamlit app structure
-st.title("Text2Text Generation App")
-st.write("Enter some text and the model will predict its sentiment:")
+st.title("Text2Text Generation App for Machine Learning Project")
+st.write("Enter some cursing sentence (toxic sentence) and the model will generate the non-toxic one:")
 
 # Input text from the user
-user_input = st.text_input("Input Text:")
+user_input = st.text_input("Input sentence:")
 
 # When the button is clicked, classify the input
 if st.button("Generate"):
     if user_input:
-        input_ids = tokenizer.encode('This is completely idiotic!', return_tensors='pt')
-        output_ids = pipe(input_ids, max_length=50, num_return_sequences=1)
-        output_text = tokenizer.decode(output_ids[0], skip_special_tokens=True)
-        #result = pipe(user_input)
-        st.write(output_text)
+        output = pipe(user_input)
+        st.write(output)
     else:
         st.write("Please enter some text.")

requirements.txt

@@ -1,3 +1,4 @@
-transformers
-streamlit
-torch  # This is for PyTorch
+streamlit==1.39.0
+transformers==4.40.1
+transformers==4.45.2
+torch