update

pages/summarization_example.py

@@ -30,9 +30,6 @@ def generate_answer(sources, model_name):
 
 summarization: """
     content = meta_prompt.format(sources=sources)
-
-    # pipe = pipeline("text-generation", model="microsoft/Phi-3-mini-4k-instruct", trust_remote_code=True, max_length=500)
-    # messages = [{"role": "user", "content": content}]
     answer = model.gen(content)
 
     return answer
@@ -56,7 +53,7 @@ def create_summarization_interface():
             model_dropdown = gr.Dropdown(choices=["gpt-3.5-turbo", "gpt-4o", "gpt-4"], label="Choose a model", value="gpt-3.5-turbo")
         
         input_text = gr.Textbox(label="Input Text", lines=10, placeholder="Enter text here...")
-        submit_button = gr.Button("Submit")
+        submit_button = gr.Button("✨ Submit ✨")
         output = gr.Markdown()
 
         example_dropdown.change(update_input, inputs=[example_dropdown], outputs=[input_text])

utils/model.py

@@ -16,7 +16,7 @@ class Model():
             device_map="auto",
         )
 
-    def gen(self, content, temp=0.0, max_length=200):
+    def gen(self, content, temp=0.0, max_length=500):
         sequences = self.pipeline(
             content,
             max_length=max_length,