nktssk commited on
Commit
6c3f51d
·
verified ·
1 Parent(s): 54bde93

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -1
app.py CHANGED
@@ -95,7 +95,11 @@ def replace_numbers_with_text_in_context(text):
95
  result.append(token)
96
  return ' '.join(result)
97
 
98
- summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
 
 
 
 
99
  translator = pipeline("translation_en_to_ru", model="Helsinki-NLP/opus-mt-en-ru")
100
  wiki = wikipediaapi.Wikipedia("Nikita", "en")
101
 
@@ -162,6 +166,11 @@ def process_image_clip(image):
162
  def process_text_clip(landmark):
163
  return process_landmark(landmark)
164
 
 
 
 
 
 
165
  with gr.Blocks() as demo:
166
  gr.Markdown("## Помощь туристу")
167
  with gr.Tabs():
@@ -175,5 +184,6 @@ with gr.Blocks() as demo:
175
  btn_txt = gr.Button("Поиск по названию")
176
  btn_img.click(fn=process_image_clip, inputs=image_input, outputs=audio_output)
177
  btn_txt.click(fn=process_text_clip, inputs=text_input, outputs=audio_output)
 
178
 
179
  demo.launch(debug=True)
 
95
  result.append(token)
96
  return ' '.join(result)
97
 
98
+ summarizer = pipeline(
99
+ "summarization",
100
+ model="sshleifer/distilbart-cnn-12-6",
101
+ tokenizer="sshleifer/distilbart-cnn-12-6"
102
+ )
103
  translator = pipeline("translation_en_to_ru", model="Helsinki-NLP/opus-mt-en-ru")
104
  wiki = wikipediaapi.Wikipedia("Nikita", "en")
105
 
 
166
  def process_text_clip(landmark):
167
  return process_landmark(landmark)
168
 
169
+ def reload_landmarks():
170
+ global landmark_titles, text_embeds
171
+ url = "https://raw.githubusercontent.com/nktssk/tourist-helper/refs/heads/main/landmarks.json"
172
+ landmark_titles = load_attractions_json(url)
173
+
174
  with gr.Blocks() as demo:
175
  gr.Markdown("## Помощь туристу")
176
  with gr.Tabs():
 
184
  btn_txt = gr.Button("Поиск по названию")
185
  btn_img.click(fn=process_image_clip, inputs=image_input, outputs=audio_output)
186
  btn_txt.click(fn=process_text_clip, inputs=text_input, outputs=audio_output)
187
+ btn_reload.click(fn=reload_landmarks, inputs=None, outputs=None)
188
 
189
  demo.launch(debug=True)