Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 from huggingface_hub import InferenceClient
 import pandas as pd
 import torch
+import math
 
 """
 For more information on `huggingface_hub` Inference API support, please check the docs: https://huggingface.co/docs/huggingface_hub/v0.22.2/en/guides/inference
@@ -35,7 +36,7 @@ def respond(
     selected_dfs = torch.load('selected_dfs.sav', map_location=torch.device('cpu'))
     for df in selected_dfs:
         df = df.dropna()
-        df = df.sample(df.shape[0].div(10))
+        df = df.sample(math.floor(df.shape[0]/10))
         for index, row in df.iterrows():
             messages.append({"role": "user", "content": row['Column1.question']})
             messages.append({"role": "assistant", "content": row['Column1.answer']})