Upload 4 files

app.txt

@@ -0,0 +1,21 @@
+import streamlit as st
+import torch
+import pickle
+
+# Load the saved model on the CPU
+model = torch.load('saved_model.pth', map_location=torch.device('cpu'))
+
+# Load the saved tokenizer
+with open('tokenizer.pkl', 'rb') as f:
+    tokenizer = pickle.load(f)
+
+st.title("Text Classification Streamlit App")
+
+input_text = st.text_input("Enter text:")
+if st.button("Predict"):
+    with torch.no_grad():
+        inputs = tokenizer(input_text, return_tensors="pt")
+        logits = model(**inputs).logits
+        predicted_class = torch.argmax(logits, dim=1).item()
+
+        st.write(f"Predicted Class: {predicted_class}")

requirements.txt

@@ -0,0 +1,3 @@
+torch
+transformers
+

saved_model (2).pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1f382c7389e4f5e8477ab48743fc6da57a5de01a1914c0a2ce242f082c3db0c1
+size 510443484

tokenizer (2).pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6488d53ece4c6b9a687dfdda95a243ffafaea3a6399c1afac457b4beb1aa8a3f
+size 2833452