Update app.py

app.py

@@ -2,13 +2,13 @@ import gradio as gr
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 
 models = {
-    "Model 1 (facebook/bart-large-cnn)": {
-        "model_name": "facebook/bart-large-cnn",
-        "description": "Model 1",
+    "Model 1 (Aiyan99/theus_concepttagger)": {
+        "model_name": "Aiyan99/theus_concepttagger",
+        "description": "Model 1: My finetuned model",
     },
-    "Model 2 (google/pegasus-multi_news)": {
-        "model_name": "google/pegasus-multi_news",
-        "description": "Model 2",
+    "Model 2 (lmsys/vicuna-13b-v1.3)": {
+        "model_name": "lmsys/vicuna-13b-v1.3",
+        "description": "Model 2: Vicuna 13B",
     },
 }
 
@@ -23,7 +23,7 @@ def summarize_text(input_text, selected_model):
     return summary
 
 custom_title = """
-<div style="color: white; text-align: center; background-color: black; padding: 20px;">
+<div style="color: black; text-align: center; background-color: white; padding: 20px;">
     <h1>MLE - Project (Tuning and Infra Project)1MLE - Project (Tuning and Infra Project)</h1>
 </div>
 """