Update app.py

app.py

@@ -1,29 +1,24 @@
 import gradio as gr
 
-# Load the model
-interface = gr.load("models/alexue4/text-normalization-ru-new")
-
-# Function to modify the behavior (example: limiting max tokens and setting examples)
-def custom_function(input_text):
-    # Custom processing here (e.g., limit max tokens)
-    max_tokens = 250  # Set your desired max token limit
-    if len(input_text.split()) > max_tokens:
-        input_text = ' '.join(input_text.split()[:max_tokens])
-    
-    # Call the original function from the interface
-    original_output = interface.predict(input_text)
-    return original_output
-
+# Define the examples
 examples = [
     ["в 2006-2010 гг. Вася учился в МГУ"],
     ["я купил iphone 10X за 14990 руб без 3-x часов полдень и т.д."]
 ]
 
-gr.Interface(
-    fn=custom_function,
-    inputs="text",
-    outputs="text",
-    examples=examples
-).launch()
+# 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)
 
+# Create the Gradio interface
+interface = gr.Interface(
+    fn=model,  # The model function to call
+    inputs=gr.inputs.Textbox(lines=2, placeholder="Enter text here..."),  # Input type
+    outputs="text",  # Output type
+    examples=examples  # Examples to display
+)
 
+# Launch the interface
+interface.launch()