Update app.py

app.py

@@ -4,6 +4,7 @@ import numpy as np
 import pandas as pd
 import pickle
 import datasets
+from datasets import Dataset, DatasetDict
 from six import BytesIO
 import gradio as gr
 from huggingface_hub import snapshot_download
@@ -30,7 +31,7 @@ def load_model():
     download_dir = snapshot_download(REPO_ID)
     saved_model_dir = os.path.join(download_dir, "saved_model")
     prediction_model = NeuralForecast.load(saved_model_dir)
-    futr_df = datasets.Dataset.from_pandas(futr_df)
+    futr_df = Dataset.from_pandas(futr_df)
     return prediction_model
 
 
@@ -46,7 +47,7 @@ def load_model():
 
 #def predict2(image_np):
 
-    results = detection_model(image_np)
+#    results = detection_model(image_np)
 
     # different object detection models have additional results
 #    result = {key:value.numpy() for key,value in results.items()}