Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -2,10 +2,11 @@ from fastapi import FastAPI, UploadFile, File
|
|
| 2 |
from typing import List
|
| 3 |
import gradio as gr
|
| 4 |
from transformers import pipeline
|
|
|
|
| 5 |
|
| 6 |
app = FastAPI(title="Eroha Summarizer PRO", version="3.6")
|
| 7 |
|
| 8 |
-
#
|
| 9 |
summarizer = None
|
| 10 |
|
| 11 |
@app.get("/")
|
|
@@ -23,7 +24,7 @@ async def summarize(files: List[UploadFile] = File(...)):
|
|
| 23 |
summary = summarizer(full_text, max_length=180, min_length=50, do_sample=False)
|
| 24 |
return {"summary": summary[0]["summary_text"]}
|
| 25 |
|
| 26 |
-
#
|
| 27 |
def gradio_summary(text):
|
| 28 |
global summarizer
|
| 29 |
if summarizer is None:
|
|
@@ -39,7 +40,12 @@ iface = gr.Interface(
|
|
| 39 |
description="AI-система для анализа и суммаризации текстов."
|
| 40 |
)
|
| 41 |
|
| 42 |
-
|
| 43 |
-
|
| 44 |
-
|
| 45 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 2 |
from typing import List
|
| 3 |
import gradio as gr
|
| 4 |
from transformers import pipeline
|
| 5 |
+
import uvicorn
|
| 6 |
|
| 7 |
app = FastAPI(title="Eroha Summarizer PRO", version="3.6")
|
| 8 |
|
| 9 |
+
# Ленивая загрузка модели
|
| 10 |
summarizer = None
|
| 11 |
|
| 12 |
@app.get("/")
|
|
|
|
| 24 |
summary = summarizer(full_text, max_length=180, min_length=50, do_sample=False)
|
| 25 |
return {"summary": summary[0]["summary_text"]}
|
| 26 |
|
| 27 |
+
# Gradio-интерфейс
|
| 28 |
def gradio_summary(text):
|
| 29 |
global summarizer
|
| 30 |
if summarizer is None:
|
|
|
|
| 40 |
description="AI-система для анализа и суммаризации текстов."
|
| 41 |
)
|
| 42 |
|
| 43 |
+
# Запуск FastAPI + Gradio (на Hugging Face)
|
| 44 |
+
@app.on_event("startup")
|
| 45 |
+
async def startup_event():
|
| 46 |
+
print("✅ Запуск Gradio интерфейса...")
|
| 47 |
+
iface.launch(share=False, server_name="0.0.0.0", server_port=7860, inbrowser=False, prevent_thread_lock=True)
|
| 48 |
+
|
| 49 |
+
# Чтобы FastAPI запустился корректно
|
| 50 |
+
if __name__ == "__main__":
|
| 51 |
+
uvicorn.run(app, host="0.0.0.0", port=7860)
|