ZennyKenny commited on
Commit
52d8051
Β·
verified Β·
1 Parent(s): 45722b4

use datasets library

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -1,10 +1,11 @@
1
  import gradio as gr
2
  from transformers import pipeline
 
3
  import pandas as pd
4
 
5
- # Load the dataset
6
- DATASET_URL = 'https://huggingface.co/datasets/ZennyKenny/demo_customer_nps/resolve/main/customer_feedback_dataset.csv'
7
- df = pd.read_csv(DATASET_URL)
8
 
9
  # Initialize the model pipeline
10
  pipe = pipeline("text-generation", model="mistralai/Mistral-Small-24B-Base-2501")
 
1
  import gradio as gr
2
  from transformers import pipeline
3
+ from datasets import load_dataset
4
  import pandas as pd
5
 
6
+ # Load the dataset from Hugging Face
7
+ ds = load_dataset('ZennyKenny/demo_customer_nps')
8
+ df = pd.DataFrame(ds['train'])
9
 
10
  # Initialize the model pipeline
11
  pipe = pipeline("text-generation", model="mistralai/Mistral-Small-24B-Base-2501")