Update app.py

app.py

@@ -37,6 +37,8 @@ dataframes = {
 #df["embedding"]=df["embedding"].apply(eval).apply(np.array)
 def reply(input, dataframe_name):
     try:
+        if dataset_name not in dataframes:
+            return "Invalid dataset selected. Please select a valid dataset."
         df = dataframes[dataframe_name]
         input = input
         input_vector = generate_embeddings(input, model_id,hf_token)