Update app.py

app.py

@@ -25,18 +25,25 @@ def inference(prompt):
   pipe = pipeline(task="text-generation",model=model,tokenizer=tokenizer,max_length=200)
   result = pipe(f"<s>[INST] {prompt} [/INST]")
   return result[0]['generated_text']
+
+INTERFACE = gr.Interface(fn=inference, inputs=[gr.Textbox(label= "Prompt", value= 'Please write about Shakuntala Devi'),
+                    
+                    outputs=gr.Text(label= "Generated Text"), title="Phi-2 fine-tuned with OpenAssistant/oasst-1 dataset using QLoRA strategy",
+                 
+                 examples = [['Write a note on Indiana Jones],]
+                        ).launch(debug=True)
     
-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")
-    examples = gr.Examples(examples=[[prompt = 'Please write about Shakuntala Devi'], [prompt = 'Write a brief note on Indiana Jones']], cache_examples=False)
-    greet_btn.click(fn=inference, inputs=prompt, outputs=output)
-
-demo.launch(debug=True)
+# 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")
+#     examples = gr.Examples(examples=[[prompt = 'Please write about Shakuntala Devi'], [prompt = 'Write a brief note on Indiana Jones']], cache_examples=False)
+#     greet_btn.click(fn=inference, inputs=prompt, outputs=output)
+
+# demo.launch(debug=True)