Update app.py

app.py

@@ -151,19 +151,27 @@
 from transformers import pipeline
 import gradio as gr
 
+# Load the pipeline once, outside of the function
+pipe = pipeline("text-generation", model="nicolaakmal/llama32-lora-finetuned-v3-fp16")
+
 def generate_response(inputs):
     messages = [
         {"role": "user", "content": inputs},
     ]
-    pipe = pipeline("text-generation", model="nicolaakmal/llama32-lora-finetuned-v3-fp16")
-    return pipe(messages)
+    # Generate the response using the pipeline
+    response = pipe(messages)
+    # Extract the generated text from the response and clean it up if necessary
+    generated_text = response[0]["generated_text"] if response else "No response generated"
+    return generated_text.strip()  # Removing any unwanted whitespace
 
 # Antarmuka Gradio
 iface = gr.Interface(
     fn=generate_response,
     inputs="text",
     outputs="text",
-    title="Chatbot LLaMA Finetuned"
+    title="Chatbot LLaMA Finetuned",
+    description="A fine-tuned LLaMA model. Enter your questions or prompts below.",
+    theme="default"  # Optional: set a theme if desired, such as 'huggingface' or 'dark'
 )
 
 # Jalankan aplikasi Gradio
@@ -173,3 +181,4 @@ iface.launch()
 
 
 
+