Update app.py

app.py

@@ -1,15 +1,14 @@
 import gradio as gr
-import torch
 from fastai.text.all import load_learner
 from huggingface_hub import hf_hub_download
 
 # Step 8: Download the model from Hugging Face and load it
 def load_model():
     try:
-        # Download the model .pth file from Hugging Face
+        # Download the model .pkl file from Hugging Face
         model_path = hf_hub_download(
             repo_id="rahul7star/fastai-rahul-text-model-v02", 
-            filename="model1.pth"
+            filename="rahul9star.pkl"
         )
         
         # Load the model using FastAI's load_learner method
@@ -20,7 +19,7 @@ def load_model():
         print(f"Error loading the model: {e}")
         return None
 
-# Load the model (this will be run at the start)
+# Load the model
 learn = load_model()
 
 # Check if the model is loaded successfully