Update app.py

app.py

@@ -15,10 +15,10 @@ tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
 tokenizer.pad_token = tokenizer.eos_token
 
 # Loading adapter (trained LORA weights)
-ckpt = '/content/drive/MyDrive/S27/results/checkpoint-500'
-model.load_adapter(ckpt)
-# adapter_path = 'checkpoint-500'
-# model.load_adapter(adapter_path)
+# ckpt = '/content/drive/MyDrive/S27/results/checkpoint-500'
+# model.load_adapter(ckpt)
+adapter_path = 'checkpoint-500'
+model.load_adapter(adapter_path)
 
 def inference(prompt):
 
@@ -27,9 +27,15 @@ def inference(prompt):
   return result[0]['generated_text']
     
 with gr.Blocks() as demo:
+
+    gr.Markdown(
+    """
+    # Phi2 trained on OpenAssistant/oasst1  dataset
+    Start typing below to see the output.
+    """)
     prompt = gr.Textbox(label="Prompt")
     output = gr.Textbox(label="Output Box")
     greet_btn = gr.Button("Generate")
-    greet_btn.click(fn=inference, inputs=prompt, outputs=output, api_name="inference")
+    greet_btn.click(fn=inference, inputs=prompt, outputs=output)
 
-demo.launch()
+demo.launch(debug=True)