jiminaa commited on
Commit
29a0404
·
1 Parent(s): 717bc81
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -4,7 +4,7 @@ from peft import PeftModel
4
  from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
5
  from threading import Thread
6
  from fastapi import FastAPI
7
- from fastapi.responses import StreamingResponse
8
  from pydantic import BaseModel
9
  import json
10
  from typing import List, Literal
@@ -289,7 +289,11 @@ with gr.Blocks(title="Language Learning Chatbot", theme=gr.themes.Soft()) as dem
289
  queue=False # Don't queue this action
290
  )
291
 
292
- app = gr.mount_gradio_app(app, demo, path="/")
 
 
 
 
293
 
294
  if __name__ == "__main__":
295
  uvicorn.run(app, host="0.0.0.0", port=7860)
 
4
  from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
5
  from threading import Thread
6
  from fastapi import FastAPI
7
+ from fastapi.responses import StreamingResponse, RedirectResponse
8
  from pydantic import BaseModel
9
  import json
10
  from typing import List, Literal
 
289
  queue=False # Don't queue this action
290
  )
291
 
292
+ app = gr.mount_gradio_app(app, demo, path="/ui")
293
+
294
+ @app.get("/")
295
+ def redirect_to_ui():
296
+ return RedirectResponse(url="/ui")
297
 
298
  if __name__ == "__main__":
299
  uvicorn.run(app, host="0.0.0.0", port=7860)