essattv commited on
Commit
a335342
verified
1 Parent(s): 0109934

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -14
app.py CHANGED
@@ -1,7 +1,11 @@
1
  import gradio as gr
2
  from transformers import AutoTokenizer, AutoModelForCausalLM
 
3
 
4
- # 艁adowanie polskiego modelu
 
 
 
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
- # Tworzenie interfejsu
40
- demo = gr.ChatInterface(
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()