Update app.py

app.py

@@ -5,11 +5,6 @@ import gradio as gr
 from booknlp.booknlp import BookNLP
 from spacy.cli import download
 
-
-#This will download the booknlp files using my huggingface backup     
-import download_missing_booknlp_models 
-
-
 # Ensure Spacy model is downloaded
 def ensure_spacy_model():
     try:
@@ -44,13 +39,14 @@ def process_book(file):
     booknlp.process(input_file, output_dir, book_id)
     
     # Zip the output folder
+    zip_file = f"{output_dir}/{book_id}_output.zip"
     shutil.make_archive(f"{output_dir}/{book_id}_output", 'zip', output_dir)
-    return f"{output_dir}/{book_id}_output.zip"
+    return zip_file
 
 # Gradio Interface
 def gradio_interface():
-    file_input = gr.inputs.File(file_types=['.txt'], label="Upload a .txt file")
-    file_output = gr.outputs.File(label="Download the output files")
+    file_input = gr.File(file_types=['.txt'], label="Upload a .txt file")
+    file_output = gr.File(label="Download the output files")
     
     gr.Interface(
         fn=process_book,