Update app.py
Browse files
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 |
-
|
99 |
-
demo.launch(
|
|
|
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()
|