Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
sachin
commited on
Commit
·
cf20ab5
1
Parent(s):
fb4d147
fix-bug
Browse files- src/server/main.py +5 -5
src/server/main.py
CHANGED
@@ -402,7 +402,7 @@ async def chat(
|
|
402 |
logger.info(f"Received prompt: {decrypted_prompt}, src_lang: {decrypted_src_lang}, user_id: {user_id}")
|
403 |
|
404 |
try:
|
405 |
-
external_url = "http://
|
406 |
payload = {
|
407 |
"prompt": decrypted_prompt,
|
408 |
"src_lang": decrypted_src_lang,
|
@@ -670,7 +670,7 @@ async def translate(
|
|
670 |
|
671 |
logger.info(f"Received translation request: {len(decrypted_sentences)} sentences, src_lang: {decrypted_src_lang}, tgt_lang: {decrypted_tgt_lang}, user_id: {user_id}")
|
672 |
|
673 |
-
external_url = "http://
|
674 |
|
675 |
payload = {
|
676 |
"sentences": decrypted_sentences,
|
@@ -767,7 +767,7 @@ async def extract_text(
|
|
767 |
start_time = time()
|
768 |
try:
|
769 |
# Call external API
|
770 |
-
external_url = f"http://
|
771 |
files = {"file": (file.filename, decrypted_content, file.content_type)}
|
772 |
|
773 |
response = requests.post(
|
@@ -877,7 +877,7 @@ async def visual_query(
|
|
877 |
"tgt_lang": decrypted_tgt_lang
|
878 |
})
|
879 |
|
880 |
-
external_url = f"http://
|
881 |
|
882 |
try:
|
883 |
files = {"file": (file.filename, decrypted_content, file.content_type)}
|
@@ -967,7 +967,7 @@ async def speech_to_speech(
|
|
967 |
encrypted_content = await file.read()
|
968 |
file_content = decrypt_data(encrypted_content, session_key)
|
969 |
files = {"file": (file.filename, file_content, file.content_type)}
|
970 |
-
external_url = f"http://
|
971 |
|
972 |
response = requests.post(
|
973 |
external_url,
|
|
|
402 |
logger.info(f"Received prompt: {decrypted_prompt}, src_lang: {decrypted_src_lang}, user_id: {user_id}")
|
403 |
|
404 |
try:
|
405 |
+
external_url = "http://155.248.245.86:7860/v1/chat"
|
406 |
payload = {
|
407 |
"prompt": decrypted_prompt,
|
408 |
"src_lang": decrypted_src_lang,
|
|
|
670 |
|
671 |
logger.info(f"Received translation request: {len(decrypted_sentences)} sentences, src_lang: {decrypted_src_lang}, tgt_lang: {decrypted_tgt_lang}, user_id: {user_id}")
|
672 |
|
673 |
+
external_url = "http://155.248.245.86:7860/v1/translate"
|
674 |
|
675 |
payload = {
|
676 |
"sentences": decrypted_sentences,
|
|
|
767 |
start_time = time()
|
768 |
try:
|
769 |
# Call external API
|
770 |
+
external_url = f"http://155.248.245.86:7861/extract-text/?page_number={page_number}"
|
771 |
files = {"file": (file.filename, decrypted_content, file.content_type)}
|
772 |
|
773 |
response = requests.post(
|
|
|
877 |
"tgt_lang": decrypted_tgt_lang
|
878 |
})
|
879 |
|
880 |
+
external_url = f"http://155.248.245.86:7860/v1/visual_query/?src_lang={decrypted_src_lang}&tgt_lang={decrypted_tgt_lang}"
|
881 |
|
882 |
try:
|
883 |
files = {"file": (file.filename, decrypted_content, file.content_type)}
|
|
|
967 |
encrypted_content = await file.read()
|
968 |
file_content = decrypt_data(encrypted_content, session_key)
|
969 |
files = {"file": (file.filename, file_content, file.content_type)}
|
970 |
+
external_url = f"http://155.248.245.86:7860/v1/speech_to_speech?language={decrypted_language}"
|
971 |
|
972 |
response = requests.post(
|
973 |
external_url,
|