add test rest api

app.py

@@ -9,6 +9,8 @@ from huggingface_hub import login
 from diffusers import FluxPipeline
 from aura_sr import AuraSR
 from deep_translator import GoogleTranslator
+from fastapi import FastAPI
+import uvicorn
 
 login(token = os.getenv('HF_TOKEN'))
 
@@ -151,6 +153,14 @@ def load_text(file_name):
     with open(file_name, 'r', encoding='utf-8') as f:
         return f.read()
 
+
+fastapi_app = FastAPI()
+
+@fastapi_app.get("/api/hello")
+def get_users():
+    return "Hello World!"
+
+
 custom_css = load_text("style.css")
 
 # Создание интерфейса Gradio
@@ -237,9 +247,12 @@ with gr.Blocks(title="3D Icon Generator", css=custom_css, theme=gr.themes.Defaul
         outputs=[output_image]
     )
 
+
+main_app = gr.mount_gradio_app(fastapi_app, app, path="/")
+
 # Запуск приложения
 if __name__ == "__main__":
-    app.launch()
+    uvicorn.run(app, host="0.0.0.0", port=7860)
 
 
 

requirements.txt

@@ -17,4 +17,6 @@ sentencepiece
 scikit-image
 rembg
 aura-sr
-deep_translator
+deep_translator
+fastapi
+uvicorn