Update app.py
Browse files
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(
|
|
|
|
|
|
|
|
|
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)
|