Update app.py

app.py

@@ -9,12 +9,16 @@ repo_dir = snapshot_download(repo_id=REPO_ID)
 
 # unzip the dir that holds the predictor
 zip_path = os.path.join(repo_dir, "autogluon_predictor_dir.zip")
-extract_to = "predictor_dir"
+extract_to = "/tmp/predictor_dir"
 if not os.path.exists(extract_to):
     with zipfile.ZipFile(zip_path, "r") as zf:
         zf.extractall(extract_to)
 
-predictor = TabularPredictor.load(extract_to, require_py_version_match=False)
+predictor = TabularPredictor.load(
+    extract_to,
+    require_py_version_match=False,
+    reset_paths=True
+)
 
 def predict_tabular(height, width, depth, page_count):
     # Validation
@@ -32,7 +36,7 @@ def predict_tabular(height, width, depth, page_count):
     df = pd.DataFrame([row])
     pred = int(predictor.predict(df)[0])
 
-    return {"Predicted Genre": pred}
+    return f"Predicted Genre Code: {int(pred)}"
 
 with gr.Blocks(title="Book Genre Predictor") as demo:
     gr.Markdown("## Predict the Genre of a Book (Numeric Labels)")
@@ -63,4 +67,4 @@ with gr.Blocks(title="Book Genre Predictor") as demo:
         outputs=out
     )
 
-
+demo.launch(share=True)