Update app.py

app.py

@@ -18,7 +18,7 @@ def wrap_text(text, width=90):
 
 def multimodal_prompt(user_input, system_prompt="You are an expert medical analyst:"):
     # Combine user input and system prompt
-    formatted_input = f"<s>[INST]{system_prompt} {user_input}[/INST]"
+    formatted_input = f"{system_prompt} {user_input}"
 
     # Encode the input text
     encodeds = tokenizer(formatted_input, return_tensors="pt", add_special_tokens=False)
@@ -55,9 +55,9 @@ class ChatBot:
     def __init__(self):
         self.history = []
 
-    def predict(self, user_input, system_prompt="You are an expert medical analyst:"):
+    def predict(self, user_input, system_prompt):
         # Combine user input and system prompt
-        formatted_input = f"<s>[INST]{system_prompt} {user_input}[/INST]"
+        formatted_input = f"{system_prompt} {user_input}"
 
         # Encode user input
         user_input_ids = tokenizer.encode(formatted_input, return_tensors="pt")
@@ -82,15 +82,14 @@ bot = ChatBot()
 
 title = "👋🏻Welcome to Tonic's Claire Chat🚀"
 description = "You can use this Space to test out the current model ([ClaireLLM](https://huggingface.co/OpenLLM-France/Claire-Mistral-7B-0.1)) or duplicate this Space and use it for any other model on 🤗HuggingFace. Join me on [Discord to build together](https://discord.gg/nXx5wbX9)."
-examples = [["Oueche Normal, Claire, ça va ou quoi?", "bonjour je m'appele Claire et je suis une assistante francophone-first conçu par openLLM"]]
-
+examples = [["[Estragon :] On va voir. Tiens. Ils prennent chacun un bout de la corde et tirent. La corde se casse. Ils manquent de tomber.", "[Vladimir] Fais voir quand même. (Estragon dénoue la corde qui maintient son pantalon.Celui-ci, beaucoup trop large, lui tombe autour des chevilles. Ils regardent la corde.) À la rigueur ça pourrait aller. Mais est-elle solide ?"]]
 iface = gr.Interface(
     fn=bot.predict,
     title=title,
     description=description,
     examples=examples,
-    inputs=["text", "text"],  # Take user input and system prompt separately
-    outputs="text",
+    inputs=[{"type": "text", "label": "Deuxieme partie d'un dialogue"}, {"type": "text", "label": "Premiere partie d'un dialogue"}],  # Labeled inputs
+    outputs={"type": "text", "label": "Claire LLM Dialogue"},  # Labeled output
     theme="ParityError/Anime"
 )