Upload prediction_compile.py

src/prediction_compile.py

@@ -120,14 +120,16 @@ def run():
         st.write(f"**Sentiment:** {sentiment_label} (Confidence: {confidence:.2f})")
 
         # Topic Modeling
-        if sentiment_label == "Negative":
-            result = topic_model_neg.transform([text])
-        else:
-            result = topic_model_pos.transform([text])
+        result = topic_model_neg.transform([text]) if sentiment_label == "Negative" else topic_model_pos.transform([text])
 
-        topics, probs = result
-        st.write(f"**Topic ID(s):** {topics}")
-        st.write(f"**Probabilities:** {probs.tolist()}")
+        if isinstance(result, tuple) and len(result) == 2:
+            topics, probs = result
+            st.write(f"**Topic ID(s):** {topics}")
+            st.write(f"**Probabilities:** {probs.tolist()}")
+        else:
+            topics = result
+            st.write(f"**Topic ID(s):** {topics}")
+            st.write("**Probabilities:** Not available")
 
 if __name__ == "__main__":
     run()