Update app.py

app.py

@@ -17,8 +17,9 @@ model_path = hf_hub_download(
 model = joblib.load(model_path)
 
 # Single prediction
-def predict_single(text):
-    return model.predict([text])[0]
+def predict_single(title, abstract):
+    df = pd.DataFrame([{"title": title, "abstract": abstract}])
+    return model.predict(df)[0]
 
 single_interface = gr.Interface(
     fn=predict_single,
@@ -27,8 +28,9 @@ single_interface = gr.Interface(
 )
 
 # Batch prediction
-def predict_batch(texts):
-    return [model.predict([t])[0] for t in texts]
+def predict_batch(df):
+    # df ya vendrá como dataframe desde gr.Dataframe
+    return model.predict(df).tolist()
 
 batch_interface = gr.Interface(
     fn=predict_batch,