traversaal-ai commited on
Commit
8750dd6
1 Parent(s): 9e2f7b0

Update run.py

Browse files
Files changed (1) hide show
  1. run.py +1 -2
run.py CHANGED
@@ -16,7 +16,7 @@ gen_kwargs = dict(
16
  top_k=30,
17
  top_p=0.9,
18
  temperature=0.8,
19
- repetition_penalty=1.05, #1.02
20
  stop= ["\n"],
21
  )
22
 
@@ -46,7 +46,6 @@ You are a chatbot. Chat in Urdu. Provide answers with your best knowledge.
46
  iface = gr.Interface(
47
  fn=generate_text,
48
  inputs=gr.Textbox(lines=2, placeholder="Enter your prompt here..."),
49
- examples = ['میں کراچی جانا چاہتا ہوں، وہاں کے کچھ بہترین مقامات کون سے ہیں؟','recipe for making Pakoras','best things to do in London'],
50
  outputs="text",
51
  title="Urdu Chatbot",
52
  description="Ask me anything in Urdu!",
 
16
  top_k=30,
17
  top_p=0.9,
18
  temperature=0.8,
19
+ repetition_penalty=1.2, #1.02
20
  stop= ["\n"],
21
  )
22
 
 
46
  iface = gr.Interface(
47
  fn=generate_text,
48
  inputs=gr.Textbox(lines=2, placeholder="Enter your prompt here..."),
 
49
  outputs="text",
50
  title="Urdu Chatbot",
51
  description="Ask me anything in Urdu!",