chansung commited on
Commit
7ca0513
1 Parent(s): d87df45

Update app/main.py

Browse files
Files changed (1) hide show
  1. app/main.py +1 -1
app/main.py CHANGED
@@ -1,6 +1,6 @@
1
  import gradio as gr
2
  from text_generation import AsyncClient
3
- from text_generation.types import Message
4
 
5
  endpoint_url = "http://127.0.0.1:8080"
6
  client = AsyncClient(endpoint_url)
 
1
  import gradio as gr
2
  from text_generation import AsyncClient
3
+ import text_generation.types.Message as Message
4
 
5
  endpoint_url = "http://127.0.0.1:8080"
6
  client = AsyncClient(endpoint_url)