kumar1907 commited on
Commit
1d21501
·
verified ·
1 Parent(s): cfdd9ee

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,5 +13,5 @@ pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
13
  def chat_with_llama(prompt):
14
  result = pipe(prompt, max_new_tokens=256, do_sample=True, temperature=0.7)
15
  return result[0]['generated_text']
16
-
17
  gr.Interface(fn=chat_with_llama, inputs="text", outputs="text", title="Venkat Assistant").launch()
 
13
  def chat_with_llama(prompt):
14
  result = pipe(prompt, max_new_tokens=256, do_sample=True, temperature=0.7)
15
  return result[0]['generated_text']
16
+
17
  gr.Interface(fn=chat_with_llama, inputs="text", outputs="text", title="Venkat Assistant").launch()