Update main.py

main.py

@@ -14,6 +14,13 @@ def t5(input):
     output = pipe_flan(input)
     return {"output": output[0]["generated_text"]}
 
+@app.get("/", response_class=HTMLResponse)
+async def streamlit_proxy(request: Request):
+    async with httpx.AsyncClient() as client:
+        streamlit_url = "http://localhost:8501/"
+        response = await client.get(streamlit_url)
+        return HTMLResponse(content=response.text, status_code=response.status_code)
+
 @app.on_event("startup")
 async def startup_event():
     # Запуск Streamlit в отдельном процессе