Update app.py

app.py

@@ -16,7 +16,7 @@ def load_model(model_name):
     model = AutoModelForSequenceClassification.from_pretrained(model_name)
     return pipeline('text-classification', model=model, tokenizer=tokenizer, truncation=True, max_length=512, top_k=4)
 
-classifier = load_model("./fine-tuned-distillberta")
+classifier = load_model("./fine_tuned_roberta-base")
 
 # Load sample essays
 with open(SAMPLE_JSON_PATH, 'r') as f:
@@ -62,7 +62,7 @@ def count_words_detection_tab(text):
 
 def generate_text_challenge_tab(): 
     global index
-    
+
     mg = gr.Button("Machine-Generated", variant="secondary", interactive=True)
     hw = gr.Button("Human-Written", variant="secondary", interactive=True)
     mh = gr.Button("Machine-Humanized", variant="secondary", interactive=True)    
@@ -115,33 +115,17 @@ css = """
 body, .gradio-container {
     font-family: Arial, sans-serif;
 }
-.gr-button {
-    background-color: #1e1e1e;
-    border: 1px solid #333333;
-    color: #ffffff;
-}
-.gr-button:hover {
-    background-color: #2e2e2e;
-}
+
 .gr-input, .gr-textarea {
-    background-color: #1f2937;
-    border: 1px solid #333333;
-    color: #ffffff;
-}
-.gr-form {
-    background-color: #1f2937;
-    border: 1px solid #333333;
 }
+
 .class-intro {
-    background-color: #1f2937;
-    border: 1px solid #333333;
     padding: 15px;
     margin-bottom: 20px;
     border-radius: 5px;
 }
 .class-intro h2 {
     margin-top: 0;
-    color: #ffffff;
 }
 .class-intro p {
     margin-bottom: 5px;