DoctorSlimm commited on
Commit
9ba385a
·
verified ·
1 Parent(s): 65bcfd2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -23,7 +23,7 @@ def greet(user):
23
  fp = snapshot_download(LLM_MODEL_ID)
24
  move_cache()
25
 
26
- model = LLM(LLM_MODEL_DIR)
27
  sampling_params = dict(
28
  temperature = 0.3,
29
  ignore_eos = False,
@@ -39,5 +39,5 @@ def greet(user):
39
  generations.append(outputs.text)
40
  return generations[0]
41
 
42
- demo = gr.Interface(fn=greet, inputs=gr.Number(), outputs=gr.Text())
43
- demo.launch()
 
23
  fp = snapshot_download(LLM_MODEL_ID)
24
  move_cache()
25
 
26
+ model = LLM(fp)
27
  sampling_params = dict(
28
  temperature = 0.3,
29
  ignore_eos = False,
 
39
  generations.append(outputs.text)
40
  return generations[0]
41
 
42
+ demo = gr.Interface(fn=greet, inputs=gr.Text(), outputs=gr.Text())
43
+ demo.launch(share=True)