osanseviero's picture
Update app.py
b1c35ff
raw
history blame contribute delete
859 Bytes
import streamlit as st
import json
import os
import requests
import socket
def start_server():
os.system("uvicorn inference_server:app --port 8080 --host 0.0.0.0 --workers 2")
st.session_state['server_started'] = True
def is_port_in_use(port):
import socket
with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
return s.connect_ex(('0.0.0.0', port)) == 0
def inference(input_text):
req = "http://0.0.0.0:8080?input_text=" + input_text
res = requests.get(req)
st.markdown(f'## Output')
st.write(json.loads(res.text))
if 'server_started' not in st.session_state:
st.session_state['server_started'] = False
if not st.session_state['server_started']:
start_server()
st.title('FastAPI Demo')
input_text = st.text_input(
label="Write something",
value="text"
)
inference(input_text)