Update app.py

app.py

@@ -85,8 +85,17 @@ model_name = "EleutherAI/gpt-neo-1.3B"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model_gptneo = AutoModelForCausalLM.from_pretrained(model_name).to(device)
 
+
+#Load dataset
+
+dataset_desc = pd.read_csv("dataset_desc.csv", delimiter=';', usecols=['Artists', 'Style', 'Description'])
+dataset_desc.columns = dataset_desc.columns.str.lower()
+style_desc = pd.read_csv("style_desc.csv", delimiter=';')  
+style_desc.columns = style_desc.columns.str.lower()
+
 # Function to enrich prompt
 def enrich_prompt(artist, style):
+    
     artist_info = dataset_desc.loc[dataset_desc['artists'].str.lower() == artist.lower(), 'description'].values
     style_info = style_desc.loc[style_desc['style'].str.lower() == style.lower(), 'description'].values