Update app.py
Browse files
app.py
CHANGED
@@ -1,7 +1,11 @@
|
|
1 |
import gradio as gr
|
2 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
|
|
3 |
|
4 |
-
#
|
|
|
|
|
|
|
5 |
model_name = "allegro/herbert-base-cased"
|
6 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
7 |
model = AutoModelForCausalLM.from_pretrained(model_name)
|
@@ -20,29 +24,22 @@ def generate_response(prompt):
|
|
20 |
return tokenizer.decode(outputs[0], skip_special_tokens=True)
|
21 |
|
22 |
def chat(message, history):
|
23 |
-
# Formatowanie historii rozmowy
|
24 |
formatted_history = ""
|
25 |
if history:
|
26 |
for human, ai in history:
|
27 |
formatted_history += f"U偶ytkownik: {human}\nAI: {ai}\n"
|
28 |
|
29 |
-
# Tworzenie promptu
|
30 |
prompt = f"{formatted_history}U偶ytkownik: {message}\nAI:"
|
31 |
-
|
32 |
-
# Generowanie odpowiedzi
|
33 |
response = generate_response(prompt)
|
34 |
-
|
35 |
-
# Usuwanie powt贸rze艅 promptu z odpowiedzi
|
36 |
-
clean_response = response.replace(prompt, "").strip()
|
37 |
-
return clean_response
|
38 |
|
39 |
-
#
|
40 |
-
demo = gr.
|
41 |
fn=chat,
|
|
|
|
|
42 |
title="Polski ChatAI",
|
43 |
-
description="Rozmawiaj ze mn膮 po polsku!"
|
44 |
-
examples=["Cze艣膰, jak si臋 masz?", "Opowiedz mi o Warszawie", "Co to jest sztuczna inteligencja?"],
|
45 |
-
theme="soft"
|
46 |
)
|
47 |
|
48 |
demo.launch()
|
|
|
1 |
import gradio as gr
|
2 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
3 |
+
import warnings
|
4 |
|
5 |
+
# Ignorowanie ostrze偶e艅 o przestarza艂ym formacie wiadomo艣ci
|
6 |
+
warnings.filterwarnings("ignore", category=UserWarning, module="gradio.components.chatbot")
|
7 |
+
|
8 |
+
# 艁adowanie modelu
|
9 |
model_name = "allegro/herbert-base-cased"
|
10 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
11 |
model = AutoModelForCausalLM.from_pretrained(model_name)
|
|
|
24 |
return tokenizer.decode(outputs[0], skip_special_tokens=True)
|
25 |
|
26 |
def chat(message, history):
|
|
|
27 |
formatted_history = ""
|
28 |
if history:
|
29 |
for human, ai in history:
|
30 |
formatted_history += f"U偶ytkownik: {human}\nAI: {ai}\n"
|
31 |
|
|
|
32 |
prompt = f"{formatted_history}U偶ytkownik: {message}\nAI:"
|
|
|
|
|
33 |
response = generate_response(prompt)
|
34 |
+
return response.replace(prompt, "").strip()
|
|
|
|
|
|
|
35 |
|
36 |
+
# Prostszy interfejs bez przyk艂ad贸w, kt贸re mog膮 powodowa膰 problemy
|
37 |
+
demo = gr.Interface(
|
38 |
fn=chat,
|
39 |
+
inputs=gr.Textbox(lines=2, placeholder="Wpisz swoje pytanie..."),
|
40 |
+
outputs="text",
|
41 |
title="Polski ChatAI",
|
42 |
+
description="Rozmawiaj ze mn膮 po polsku!"
|
|
|
|
|
43 |
)
|
44 |
|
45 |
demo.launch()
|