Update app.py

app.py

@@ -5,7 +5,7 @@ from inference.main import MultiModalPhi2
 messages = []
 
 multimodal_phi2 = MultiModalPhi2(
-    modelname_or_path="Navyabhat/Llava-Phi2",
+    modelname_or_path="RaviNaik/Llava-Phi2",
     temperature=0.2,
     max_new_tokens=1024,
     device="cpu",
@@ -61,85 +61,41 @@ def run(history, text, image, audio_upload, audio_mic):
     return history, None, None, None, None
 
 
-# Custom styling
-interface_style = {
-    "box": {
-        "backgroundColor": "#f9f9f9",
-        "padding": "20px",
-        "borderRadius": "10px",
-        "boxShadow": "0 0 10px rgba(0, 0, 0, 0.1)",
-    },
-    "button": {
-        "backgroundColor": "#4caf50",
-        "color": "#fff",
-        "padding": "10px",
-        "border": "none",
-        "borderRadius": "5px",
-        "cursor": "pointer",
-    },
-    "textbox": {
-        "width": "100%",
-        "padding": "10px",
-        "marginBottom": "10px",
-        "boxSizing": "border-box",
-    },
-    "image": {
-        "width": "100%",
-        "marginBottom": "10px",
-    },
-    "audio": {
-        "width": "100%",
-        "marginBottom": "10px",
-    },
-    "chatbox": {
-        "height": "550px",
-        "backgroundColor": "#f0f0f0",
-        "borderRadius": "5px",
-        "padding": "10px",
-        "overflowY": "auto",
-    },
-}
-
 with gr.Blocks() as demo:
-    gr.Markdown("## MultiModal Phi2 Model Pretraining and Finetuning from Scratch")
+    gr.Markdown("## MulitModal Phi2 Model Pretraining and Finetuning from Scratch")
 
     with gr.Row():
         with gr.Column(scale=4):
-            with gr.Box(style=interface_style["box"]):
+            
+            with gr.Box():
                 with gr.Row():
+                    # Adding a Textbox with a placeholder "write prompt"
                     prompt = gr.Textbox(
-                        placeholder="Ask anything",
-                        lines=2,
-                        label="Query",
-                        value=None,
-                        style=interface_style["textbox"],
+                        placeholder="Ask anything", lines=2, label="Query", value=None
                     )
+                
                 with gr.Row():
-                    image = gr.Image(
-                        type="filepath", value=None, style=interface_style["image"]
-                    )
+                    # Adding image
+                    image = gr.Image(type="filepath", value=None)
+                
                 with gr.Row():
-                    audio_upload = gr.Audio(
-                        source="upload", type="filepath", style=interface_style["audio"]
-                    )
+                    # Add audio
+                    audio_upload = gr.Audio(source="upload", type="filepath")
                     audio_mic = gr.Audio(
-                        source="microphone",
-                        type="filepath",
-                        format="mp3",
-                        style=interface_style["audio"],
+                        source="microphone", type="filepath", format="mp3"
                     )
 
         with gr.Column(scale=8):
-            with gr.Box(style=interface_style["box"]):
+            with gr.Box():
                 with gr.Row():
                     chatbot = gr.Chatbot(
                         avatar_images=("🧑", "🤖"),
                         height=550,
-                        style=interface_style["chatbox"],
                     )
                 with gr.Row():
-                    submit = gr.Button(value = 'Submit', style=interface_style["button"])
-                    clear = gr.Button(value="Clear", style=interface_style["button"])
+                    # Adding a Button
+                    submit = gr.Button()
+                    clear = gr.Button(value="Clear")
 
     submit.click(
         add_content,