Update app.py

app.py

@@ -22,7 +22,9 @@ def generate_dialogue(input_text):
 
   pipe = pipeline(task="text-generation",model=model,tokenizer=tokenizer,max_length=200)
   result = pipe(f"<s>[INST] {input_text} [/INST]")
-  return result[0]['generated_text']
+
+  return_answer = (result[0]['generated_text']).split('[s]')[1].split('[/s]')[0]
+  return return_answer
     
 HTML_TEMPLATE = """
   <style>
@@ -77,7 +79,7 @@ HTML_TEMPLATE = """
     <div class="artifact large"></div>
     <div class="artifact large"></div>
     <!-- Content -->
-    <h1>CHAT with fine tuned Phi-2 LLM</h1>
+    <h1>CHAT with Fine Tuned Phi-2 LLM</h1>
     <p>Generate dialogue for given some initial prompt for context.</p>
     <p>Model: Phi-2 (https://huggingface.co/microsoft/phi-2),  Dataset: oasst1 (https://huggingface.co/datasets/OpenAssistant/oasst1) </p>
 """