Update app.py

app.py

@@ -1,8 +1,12 @@
 import gradio as gr
-from fastai.text.all import load_learner
+from fastai.text.all import load_learner, TextBlock, CategoryBlock
 from huggingface_hub import hf_hub_download
 
-# Step 1: Load the model from Hugging Face
+# Step 1: Redefine Custom Functions
+def get_x(x): return x['input']
+def get_y(x): return x['output']
+
+# Step 2: Load the model from Hugging Face
 def load_model():
     try:
         # Download the .pkl file from Hugging Face
@@ -26,7 +30,7 @@ learn = load_model()
 if learn is None:
     raise ValueError("Failed to load the model")
 
-# Step 2: Define the Gradio Interface
+# Step 3: Define the Gradio Interface
 def predict(input_text):
     try:
         # Get prediction from the model
@@ -35,5 +39,5 @@ def predict(input_text):
     except Exception as e:
         return f"Error during prediction: {e}"
 
-# Step 3: Create Gradio Interface
+# Step 4: Create Gradio Interface
 gr.Interface(fn=predict, inputs="text", outputs="text").launch()