Update app.py

app.py

@@ -1,36 +1,29 @@
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 import gradio as gr
-from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
 
-model_id = "tiiuae/falcon-rw-1b"  # Szybszy i sensowniejszy model
-
-tokenizer = AutoTokenizer.from_pretrained(model_id)
-model = AutoModelForCausalLM.from_pretrained(model_id, trust_remote_code=True)
-generator = pipeline("text-generation", model=model, tokenizer=tokenizer)
-
-chat_history = []
-
-def chat(user_input):
-    global chat_history
-    chat_history.append(f"User: {user_input}")
-    prompt = "\n".join(chat_history) + "\nAI:"
-    
-    response = generator(prompt, max_new_tokens=80, do_sample=True, top_k=40, top_p=0.9, temperature=0.7)[0]["generated_text"]
-    reply = response[len(prompt):].strip().split("\nUser:")[0]
-    
-    chat_history.append(f"AI: {reply}")
-    return reply
-
-with gr.Blocks() as demo:
-    gr.Markdown("## 🤖 Polski Chatbot AI – Szybki i Inteligentny")
-    chatbox = gr.Chatbot()
-    msg = gr.Textbox(label="Twoja wiadomość")
-    send_btn = gr.Button("Wyślij")
-
-    def respond(user_input, history=[]):
-        reply = chat(user_input)
-        history.append((user_input, reply))
-        return history, ""
-
-    send_btn.click(respond, inputs=[msg, chatbox], outputs=[chatbox, msg])
-
-demo.launch()
+# Załaduj model
+model_name = "google/mt5-small"
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
+
+# Funkcja odpowiadająca na pytanie
+def answer_question(question):
+    input_text = f"Pytanie: {question} Odpowiedź:"
+    inputs = tokenizer.encode(input_text, return_tensors="pt")
+    output = model.generate(
+        inputs,
+        max_new_tokens=60,
+        do_sample=False,
+        temperature=0.3,
+        top_p=0.95
+    )
+    return tokenizer.decode(output[0], skip_special_tokens=True)
+
+# Gradio UI
+gr.Interface(
+    fn=answer_question,
+    inputs=gr.Textbox(lines=2, placeholder="Zadaj pytanie..."),
+    outputs=gr.Textbox(),
+    title="🤖 Polski Chatbot AI",
+    description="Zadaj pytanie po polsku, a chatbot udzieli sensownej odpowiedzi"
+).launch()