Update app.py

app.py

@@ -18,23 +18,28 @@ import gradio as gr
 from transformers import AutoModel, AutoTokenizer
 import torch
 from sklearn.model_selection import train_test_split
-from google.colab import files
 
-# Upload the dataset if running in Google Colab
-def upload_dataset():
-    uploaded = files.upload()  # This will prompt the file upload
-    file_name = list(uploaded.keys())[0]
-    file_path = f'/content/{file_name}'
-    return file_path
+# Function to convert a list to a DataFrame
+def list_to_dataframe(data_list):
+    # Convert the list to a DataFrame (assuming it's a list of dicts or tuples)
+    df = pd.DataFrame(data_list)
+    return df
 
-# Load your dataset
-def load_dataset():
-    file_path = '/content/Valid-part-2.xlsx'  # Default path if the file is uploaded manually to Colab
+# Load your dataset from a file
+def load_dataset(file_path=None):
+    if file_path is None:
+        file_path = '/content/Valid-part-2.xlsx'  # Default path if the file is uploaded manually to Colab
 
     # Check if the file exists
-    if not os.path.exists(file_path):
-        print(f"File not found at '{file_path}', prompting file upload...")
-        file_path = upload_dataset()  # Upload if not found
+    if file_path and not os.path.exists(file_path):
+        print(f"File not found at '{file_path}', using default list data...")
+        # Fallback to a default list if file is not found
+        default_data = [
+            {'text': 'Example sentence 1', 'label': 'label1'},
+            {'text': 'Example sentence 2', 'label': 'label2'},
+            # Add more example data as needed
+        ]
+        return list_to_dataframe(default_data)
     
     try:
         df = pd.read_excel(file_path)
@@ -78,8 +83,8 @@ def predict(input_text):
     return outputs.last_hidden_state
 
 # Build the Gradio interface
-def build_interface():
-    df = load_dataset()  # Load your dataset
+def build_interface(file_path=None):
+    df = load_dataset(file_path)  # Load your dataset
     if df is None:
         return None
 
@@ -95,7 +100,9 @@ def build_interface():
 
 # Run the Gradio interface
 if __name__ == "__main__":
-    iface = build_interface()
+    # You can specify a file_path here if you have a specific file to use
+    file_path = None  # Change this to your specific file path if needed
+    iface = build_interface(file_path=file_path)
     if iface:
         iface.launch()
     else:
@@ -103,3 +110,4 @@ if __name__ == "__main__":
 
 
 
+