Yermek68 commited on
Commit
b1a1fe2
·
verified ·
1 Parent(s): 7d41bbc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -8
app.py CHANGED
@@ -1,26 +1,25 @@
1
  from transformers import pipeline
2
  import gradio as gr
3
 
4
- # Ленивая инициализация модели
5
  summarizer = None
6
 
7
  def summarize_text(text):
8
  global summarizer
9
  if summarizer is None:
10
  summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
11
- summary = summarizer(text, max_length=180, min_length=50, do_sample=False)
12
- return summary[0]["summary_text"]
13
 
14
- # Интерфейс Gradio
15
  iface = gr.Interface(
16
  fn=summarize_text,
17
  inputs=gr.Textbox(lines=10, label="Введите текст для суммаризации"),
18
  outputs="text",
19
  title="Eroha Summarizer PRO",
20
- description="AI-система для анализа и суммаризации текстов на русском и английском языках."
21
  )
22
 
23
- # 👉 Главное отличие:
24
- # Hugging Face Spaces не нужно запускать вручную через iface.launch().
25
- # Он сам вызовет интерфейс, если ты вернёшь переменную app:
26
  app = iface
 
1
  from transformers import pipeline
2
  import gradio as gr
3
 
4
+ # Ленивая загрузка модели (чтобы не занимала память до первого вызова)
5
  summarizer = None
6
 
7
  def summarize_text(text):
8
  global summarizer
9
  if summarizer is None:
10
  summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
11
+ result = summarizer(text, max_length=180, min_length=50, do_sample=False)
12
+ return result[0]["summary_text"]
13
 
14
+ # Gradio-интерфейс
15
  iface = gr.Interface(
16
  fn=summarize_text,
17
  inputs=gr.Textbox(lines=10, label="Введите текст для суммаризации"),
18
  outputs="text",
19
  title="Eroha Summarizer PRO",
20
+ description="🧠 AI-система Ерохи для анализа и суммаризации текстов (EN/RU)."
21
  )
22
 
23
+ # 👉 Hugging Face Spaces автоматически ищет переменную `app`
24
+ # Не нужно использовать iface.launch()
 
25
  app = iface