Spaces:
Sleeping
Sleeping
fixing ui
Browse files
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)
|