RashiAgarwal commited on
Commit
d497b84
1 Parent(s): 5c0dd13

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -26,11 +26,11 @@ def inference(prompt):
26
  result = pipe(f"<s>[INST] {prompt} [/INST]")
27
  return result[0]['generated_text']
28
 
29
- INTERFACE = gr.Interface(fn=inference, inputs=[gr.Textbox(label= "Prompt", value= 'write a note on Albert Einstein')],
30
 
31
- outputs=gr.Text(label= "Generated Text"), title="Phi-2 fine-tuned with OpenAssistant/oasst-1 dataset using QLoRA strategy",
32
 
33
- examples = [['Tell me about Amitabh Bachchan'],]
34
  ).launch(debug=True)
35
 
36
  # with gr.Blocks() as demo:
 
26
  result = pipe(f"<s>[INST] {prompt} [/INST]")
27
  return result[0]['generated_text']
28
 
29
+ INTERFACE = gr.Interface(fn=inference, inputs=[gr.Textbox(label= "Prompt", value= 'what should we do to save time')],
30
 
31
+ outputs=gr.Text(label= "Generated Text"), title="Language Model Phi-2 fine-tuned with OpenAssistant/oasst-1 dataset using QLoRA strategy",
32
 
33
+ examples = [['explain transpiration in plants'],]
34
  ).launch(debug=True)
35
 
36
  # with gr.Blocks() as demo: