Update app.py

app.py

@@ -79,7 +79,7 @@ MAX_SEQUENCE_LENGTH = 500
 # DATASET_REPO_URL = "https://huggingface.co/datasets/Seetha/Visualization"
 # DATA_FILENAME = "level2.json"
 #DATA_FILE = os.path.join("data", DATA_FILENAME)
-DATASET_REPO_URL = "https://huggingface.co/datasets/Seetha/visual_files"
+DATASET_REPO_URL = "https://huggingface.co/datasets/KGBrain/visual_files"
 DATA_FILENAME = "detailedResults.json"
 DATA_FILENAME1 = "level2.json"
 
@@ -519,10 +519,10 @@ def main():
             'value': int(df_tab.loc[row, col])
         })
 
-  HfApi().delete_file(path_in_repo = DATA_FILENAME1 ,repo_id = 'Seetha/visual_files',token= HF_TOKEN,repo_type='dataset')
+  HfApi().delete_file(path_in_repo = DATA_FILENAME1 ,repo_id = 'KGBrain/visual_files',token= HF_TOKEN,repo_type='dataset')
   #st.write('file-deleted')
   fs = HfFileSystem(token=HF_TOKEN)
-  with fs.open('datasets/Seetha/visual_files/level2.json', 'w') as f:
+  with fs.open('datasets/KGBrain/visual_files/level2.json', 'w') as f:
     json.dump(json_data, f)
   
   df_final1.to_csv('predictions.csv')
@@ -541,9 +541,9 @@ def main():
   # # Convert the list of dictionaries to JSON
   json_data = json.dumps(data_list)
 
-  HfApi().delete_file(path_in_repo = DATA_FILENAME ,repo_id = 'Seetha/visual_files',token= HF_TOKEN,repo_type='dataset')
+  HfApi().delete_file(path_in_repo = DATA_FILENAME ,repo_id = 'KGBrain/visual_files',token= HF_TOKEN,repo_type='dataset')
   #st.write('file2-deleted')
-  with fs.open('datasets/Seetha/visual_files/detailedResults.json','w') as fi:
+  with fs.open('datasets/KGBrain/visual_files/detailedResults.json','w') as fi:
     #data = json.load(fi)
     fi.write(json_data)
 
@@ -561,7 +561,7 @@ def main():
   with st.container():
     
     st.download_button(label="Download the result table",data=buffer,file_name="t2cg_outputs.xlsx",mime="application/vnd.ms-excel")
-    st.markdown('<a href="https://huggingface.co/spaces/Seetha/visual-knowledgegraph" target="_blank">Click this link in a separate tab to view knowledge graph</a>', unsafe_allow_html=True)
+    st.markdown('<a href="https://huggingface.co/spaces/KGBrain/visual-knowledgegraph" target="_blank">Click this link in a separate tab to view knowledge graph</a>', unsafe_allow_html=True)
     # st.download_button(label="Download the detailed result table_csv",data=csv1,file_name='results.csv',mime='text/csv')
     # st.download_button(label="Download the result table_csv",data=csv2,file_name='final_data.csv',mime='text/csv')