sandz7 commited on
Commit
a43653a
Β·
verified Β·
1 Parent(s): 6c232c6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -29,7 +29,7 @@ processor = AutoProcessor.from_pretrained(model_id)
29
  # model.generation_config.eos_token_id = processor.tokenizer.eos_token_id
30
  model.generation_config.eos_token_id = 128009
31
 
32
- @spaces.GPU
33
  def bot_streaming(message, history):
34
  print(message)
35
  if message["files"]:
@@ -95,5 +95,5 @@ with gr.Blocks(fill_height=True) as demo:
95
  textbox=chat_input,
96
  )
97
 
98
- demo.queue(api_open=False)
99
- demo.launch(show_api=False, share=False)
 
29
  # model.generation_config.eos_token_id = processor.tokenizer.eos_token_id
30
  model.generation_config.eos_token_id = 128009
31
 
32
+ @spaces.GPU(duration=120)
33
  def bot_streaming(message, history):
34
  print(message)
35
  if message["files"]:
 
95
  textbox=chat_input,
96
  )
97
 
98
+ if __name__ == "__main__":
99
+ demo.launch()