Update app.py

Added a theme and temperature

app.py

@@ -33,10 +33,10 @@ pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
 # Definimos el eos token para el modelo
 eos_token = tokenizer("<|im_end|>",add_special_tokens=False)["input_ids"][0]
 
-def generate_inference(instruction, input):
+def generate_inference(instruction, input, temperature):
     prompt = pipe.tokenizer.apply_chat_template([{"role": "user", 
                                                   "content": f"{instruction}/n{input}"}], tokenize=False, add_generation_prompt=True)
-    outputs = pipe(prompt, max_new_tokens=256, do_sample=True, num_beams=1, temperature=0.2, top_k=50, top_p=0.95, 
+    outputs = pipe(prompt, max_new_tokens=256, do_sample=True, num_beams=1, temperature=float(temperature), top_k=50, top_p=0.95, 
                    max_time= 300, eos_token_id=eos_token)
     return outputs[0]['generated_text'][len(prompt):].strip()
 
@@ -75,12 +75,15 @@ def clean_output():
 def clean_all():
         return "¿Quién es un solicitante de asilo/ protección internacional?","", "" #inp, cont, out
 
-with gr.Blocks(theme="ParityError/LimeFace", title="QA Legal Refugiados") as demo:
+with gr.Blocks(theme="snehilsanyal/scikit-learn", title="QA Legal Refugiados") as demo:
     title="Question Answering - Legal Refugiados v1.0"
     
     gr.HTML("""
+<p align="center">
+    <img src="/file=images/markus-winkler-Je1MDuITTF4-unsplash.jpg" width="300" height="150">
+</p>            
 <h1 align="center">Question Answering - Legal Refugiados v1.0</h1>
-<h2 align="center">Apoyo para dar información y responder consultas acerca de las leyes y procedimientos de actuación y protección a asiliados políticos y refugiados.</h2>
+<h2 align="center">Apoyo para informar y responder a consultas acerca de procedimientos y leyes de actuación y protección a asiliados y refugiados.</h2>
 """)
 
     inp = gr.Textbox(label="🌐 Pregunta a resolver",
@@ -88,31 +91,37 @@ with gr.Blocks(theme="ParityError/LimeFace", title="QA Legal Refugiados") as dem
             value="¿Quién es un solicitante de asilo/ protección internacional?",
             interactive=True,
     )
-    context=gr.Textbox(label="Contexto",
+    context=gr.Textbox(label="📰 Contexto",
             info="Introduce el contexto de la pregunta",
             placeholder="Contexto de la pregunta",
-            interactive=False,
+            interactive=True,
     )
     
-    out = gr.Textbox(label="📰 Respuesta e información",
+    out = gr.Textbox(label="🚀 Respuesta e información",
             interactive=False,
             placeholder="Aquí aparecerá la respuesta o información solicitada",
     )
-    exam_box = gr.Examples(examples, 
+    inp.change(clean_output, inputs=[], outputs=out)
+    
+    with gr.Row():
+        with gr.Column(scale=3, min_width=600):
+            exam_box = gr.Examples(examples, 
                            inp, 
                            context,
                            fn=get_context,
                            run_on_click= True,
-                           label= "Ejemplos de cuestiones. Se les asignara el contexto automáticamente")
-    inp.change(clean_output, inputs=[], outputs=out)
-    with gr.Row():
-        inference_btn = gr.Button("Responder")
-        inference_btn.click(fn=generate_inference, inputs=[inp, context], outputs=out)
-        clean_context_btn = gr.Button("Limpiar contexto")
-        clean_context_btn.click(fn=clean_output, inputs=[], outputs=context)
-        clean_btn = gr.Button("Limpiar todo")
-        clean_btn.click(fn=clean_all, inputs=[], outputs=[inp, context, out])
+                           label= "📌 Ejemplos de cuestiones. Se les asignara el contexto automáticamente")
+
+        with gr.Column(scale=1, min_width=200):
+            temperature=gr.Slider(minimum=0.2, maximum=1, step=0.1, value=0.3, label="🌡 Temperature", visible=True)
+            inference_btn = gr.Button("Responder")
+            inference_btn.click(fn=generate_inference, inputs=[inp, context, temperature], outputs=out)
+            clean_context_btn = gr.Button("Limpiar contexto")
+            clean_context_btn.click(fn=clean_output, inputs=[], outputs=context)
+            clean_btn = gr.Button("Limpiar todo")
+            clean_btn.click(fn=clean_all, inputs=[], outputs=[inp, context, out])
+    
 
 #demo.queue(max_size=None)
-demo.launch(debug=True, share=False)
+demo.launch(debug=True, share=False, allowed_paths=["images/markus-winkler-Je1MDuITTF4-unsplash.jpg"])