Update app.py

.gitignore

@@ -0,0 +1,2 @@
+Untitled.ipynb
+.ipynb_checkpoints/Untitled-checkpoint.ipynb

app.py

@@ -1,28 +1,20 @@
-from transformers import GPT2LMHeadModel, GPT2TokenizerFast
+from transformers import pipeline
 import streamlit as st
 
-# Load pre-trained GPT-2 model and tokenizer
-model_name = "gpt2"  # You can try other GPT-2 variants as well
-tokenizer = GPT2TokenizerFast.from_pretrained(model_name)
-model = GPT2LMHeadModel.from_pretrained(model_name, pad_token_id=tokenizer.eos_token_id)
+model_name = "FPRT/bert-fine-tuned-stepkids-classification"  
+pipe = pipeline(model=model_name)
 
-st.title("Theme Generation App")
+def main():
+    st.title("Hugging Face Model Demo")
 
-# Create a textarea for user input
-user_text = st.text_area("Enter Text:", "Type here...")
+    # Create an input text box
+    input_text = st.text_input("Enter your text", "")
 
-if st.button("Generate Themes"):
-    # Construct a prompt for thematic generation
-    prompt = f"Reply with at most seven possible themes detected in the text: {user_text}\nThemes:"
+    # Create a button to trigger model inference
+    if st.button("Analyze"):
+        # Perform inference using the loaded model
+        result = pipe(input_text)
+        st.write("Prediction:", result[0]['label'], "| Score:", result[0]['score'])
 
-    # Tokenize the prompt
-    input_ids = tokenizer.encode(prompt, return_tensors="pt")
-
-    # Generate themes using GPT-2
-    output = model.generate(input_ids, max_new_tokens=10, num_beams=5, no_repeat_ngram_size=2, top_k=50, top_p=0.95, temperature=0.5)
-
-    # Decode the generated themes
-    generated_themes = tokenizer.decode(output[0], skip_special_tokens=True)
-    
-    # Display the generated themes
-    st.success(f"Generated Themes: {generated_themes}")
+if __name__ == "__main__":
+    main()

requirements.txt

@@ -1,59 +1 @@
-altair==5.1.2
-attrs==23.1.0
-blinker==1.7.0
-cachetools==5.3.2
-certifi==2023.7.22
-charset-normalizer==3.3.2
-click==8.1.7
-colorama==0.4.6
-filelock==3.13.1
-fsspec==2023.10.0
-gitdb==4.0.11
-GitPython==3.1.40
-huggingface-hub==0.17.3
-idna==3.4
-importlib-metadata==6.8.0
-Jinja2==3.1.2
-jsonschema==4.19.2
-jsonschema-specifications==2023.7.1
-markdown-it-py==3.0.0
-MarkupSafe==2.1.3
-mdurl==0.1.2
-mpmath==1.3.0
-networkx==3.2.1
-numpy==1.26.1
-packaging==23.2
-pandas==2.1.2
-Pillow==10.1.0
-protobuf==4.25.0
-pyarrow==14.0.0
-pydeck==0.8.1b0
-Pygments==2.16.1
-python-dateutil==2.8.2
-pytz==2023.3.post1
-PyYAML==6.0.1
-referencing==0.30.2
-regex==2023.10.3
-requests==2.31.0
-rich==13.6.0
-rpds-py==0.12.0
-safetensors==0.4.0
-six==1.16.0
-smmap==5.0.1
-streamlit==1.28.1
-sympy==1.12
-tenacity==8.2.3
-tokenizers==0.14.1
-toml==0.10.2
-toolz==0.12.0
-torch==2.1.0
-tornado==6.3.3
-tqdm==4.66.1
-transformers==4.35.0
-typing_extensions==4.8.0
-tzdata==2023.3
-tzlocal==5.2
-urllib3==2.0.7
-validators==0.22.0
-watchdog==3.0.0
-zipp==3.17.0
+streamlit-tags