Yermek68 commited on
Commit
980abe2
·
verified ·
1 Parent(s): b758f58

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -6
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
- # ============ Gradio UI ============
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
- @app.get("/gradio")
43
- def launch_gradio():
44
- iface.launch(share=False, server_name="0.0.0.0", server_port=7860)
45
- return {"message": "Gradio interface запущен!"}
 
 
 
 
 
 
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)