Update app.py

app.py

@@ -1,4 +1,5 @@
 import gradio as gr
+from transformers import pipeline
 
 # Define the examples
 examples = [
@@ -6,15 +7,16 @@ examples = [
     ["я купил iphone 10X за 14990 руб без 3-x часов полдень и т.д."]
 ]
 
-# Load the model with arguments including max_length
-model = gr.load("models/alexue4/text-normalization-ru-new", 
-                src="https://huggingface.co/alexue4/text-normalization-ru-new", 
-                description="A model for text normalization in Russian", 
-                max_length=512)
+# Load the model using Hugging Face pipeline
+model = pipeline("text2text-generation", model="alexue4/text-normalization-ru-new")
+
+# Define the function to normalize text with max_length parameter
+def normalize_text(input_text):
+    return model(input_text, max_length=512)[0]['generated_text']
 
 # Create the Gradio interface
 interface = gr.Interface(
-    fn=model,  # The model function to call
+    fn=normalize_text,  # The function to call
     inputs=gr.inputs.Textbox(lines=2, placeholder="Enter text here..."),  # Input type
     outputs="text",  # Output type
     examples=examples  # Examples to display