Update app.py

app.py

@@ -1,21 +1,17 @@
 
 import gradio as gr
-from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
+from transformers import T5Tokenizer, MT5ForConditionalGeneration
 
-# Použijeme multijazyčný model MT5
 model_name = "google/mt5-base"
-tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
+tokenizer = T5Tokenizer.from_pretrained(model_name, use_fast=False)
+model = MT5ForConditionalGeneration.from_pretrained(model_name)
 
-# Definícia funkcie na humanizáciu
 def humanize(text):
-    # Pridáme jednoduchý prompt – MT5 používa implicitné inštrukcie
     prompt = "Parafrázuj tento text prirodzene a ľudsky: " + text
     inputs = tokenizer(prompt, return_tensors="pt", truncation=True, padding=True)
     outputs = model.generate(**inputs, max_length=512, num_beams=4)
     return tokenizer.decode(outputs[0], skip_special_tokens=True)
 
-# Gradio rozhranie
 gr.Interface(
     fn=humanize,
     inputs=gr.Textbox(lines=10, label="Vstupný text (česky alebo slovensky)"),
@@ -23,3 +19,4 @@ gr.Interface(
     title="Humanizer CZ/SK (MT5)",
     description="Prepíše český alebo slovenský text do prirodzenejšej formy pomocou multijazyčného modelu MT5."
 ).launch()
+