Upload app.py

app.py

@@ -0,0 +1,26 @@
+import streamlit as st
+import requests
+import json
+
+def main():
+
+    st.title("FastAPI - Streamlit Integration")
+
+    # Эндпоинт для классификации текста
+    text_input = st.text_input("Enter text for classification:")
+    if st.button("Classify Text"):
+        response = requests.post("http://127.0.0.1:8000/clf_text", json={"text": text_input})
+        result = response.json()
+        st.success(f"Classification result: {result['prediction']}")
+
+    # Эндпоинт для классификации изображений
+    image_path = st.file_uploader("Upload an image for classification:", type=["jpg", "png"])
+    if image_path is not None:
+        if st.button("Classify Image"):
+            files = {"file": image_path.read()}
+            response = requests.post("http://127.0.0.1:8000/classify", files=files)
+            result = response.json()
+            st.success(f"Classification result: {result['prediction']}")
+
+if __name__ == '__main__':
+    main()