Update app.py

app.py

@@ -69,7 +69,8 @@ def initialize_model_and_tokenizer(model_name, num_labels):
     except Exception as e:
         st.error(f"Error initializing model and tokenizer: {str(e)}")
         return None, None
-def create_dataset(data, tokenizer, max_length):
+
+def create_dataset(data, tokenizer, max_length, num_labels):
     texts = [item.get('prompt', '') for item in data if item.get('prompt')]
     labels = [item.get('label', 0) for item in data if item.get('prompt')]
 
@@ -94,8 +95,7 @@ def create_dataset(data, tokenizer, max_length):
         'labels': labels
     })
     return dataset
-
-
+    
 def main():
     st.title("Appointment Scheduling Platform")