Jithendra-k commited on
Commit
50fcb03
·
verified ·
1 Parent(s): b00bd5e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ def getLLamaresponse(input_text):
18
  pipe = pipeline(task="text-generation", model=model, tokenizer=tokenizer, max_length=50, do_sample=True)
19
  result = pipe(f"<s>[INST] {input_text} [/INST]")
20
  print(result[0]['generated_text'])
21
- return result[0]['generated_text']
22
 
23
  st.set_page_config(page_title="Generate Keywords from User Queries",
24
  page_icon='🤖',
 
18
  pipe = pipeline(task="text-generation", model=model, tokenizer=tokenizer, max_length=50, do_sample=True)
19
  result = pipe(f"<s>[INST] {input_text} [/INST]")
20
  print(result[0]['generated_text'])
21
+ return "Hello World!"
22
 
23
  st.set_page_config(page_title="Generate Keywords from User Queries",
24
  page_icon='🤖',