Spaces:
Build error
Build error
pengdaqian
commited on
Commit
·
5475e93
1
Parent(s):
50ddfa2
add more
Browse files
main.py
CHANGED
|
@@ -2,7 +2,7 @@ import requests
|
|
| 2 |
import uvicorn
|
| 3 |
from fastapi import BackgroundTasks, FastAPI
|
| 4 |
import model
|
| 5 |
-
from ocr import
|
| 6 |
|
| 7 |
app = FastAPI()
|
| 8 |
|
|
|
|
| 2 |
import uvicorn
|
| 3 |
from fastapi import BackgroundTasks, FastAPI
|
| 4 |
import model
|
| 5 |
+
from ocr import ocr, get_ocr_model, init_chatgpt
|
| 6 |
|
| 7 |
app = FastAPI()
|
| 8 |
|
ocr.py
CHANGED
|
@@ -33,6 +33,7 @@ def init_chatgpt():
|
|
| 33 |
"content": "You are a ocr bot, Collating the following ocr result output to turn them into dialog context",
|
| 34 |
},
|
| 35 |
]
|
|
|
|
| 36 |
|
| 37 |
|
| 38 |
def ocr(reader, chatbot, file_path: str):
|
|
|
|
| 33 |
"content": "You are a ocr bot, Collating the following ocr result output to turn them into dialog context",
|
| 34 |
},
|
| 35 |
]
|
| 36 |
+
return chatbot
|
| 37 |
|
| 38 |
|
| 39 |
def ocr(reader, chatbot, file_path: str):
|