Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -4,6 +4,7 @@ from fastapi import FastAPI, HTTPException
|
|
4 |
from pydantic import BaseModel
|
5 |
from typing import List
|
6 |
import torch
|
|
|
7 |
from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
|
8 |
from IndicTransToolkit import IndicProcessor
|
9 |
import uvicorn
|
@@ -117,8 +118,11 @@ def main():
|
|
117 |
target_lang=target_languages[target_lang]
|
118 |
)
|
119 |
|
120 |
-
#
|
121 |
-
|
|
|
|
|
|
|
122 |
|
123 |
# Display result
|
124 |
st.success("Translation:")
|
|
|
4 |
from pydantic import BaseModel
|
5 |
from typing import List
|
6 |
import torch
|
7 |
+
import asyncio
|
8 |
from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
|
9 |
from IndicTransToolkit import IndicProcessor
|
10 |
import uvicorn
|
|
|
118 |
target_lang=target_languages[target_lang]
|
119 |
)
|
120 |
|
121 |
+
# Create event loop and run translation
|
122 |
+
loop = asyncio.new_event_loop()
|
123 |
+
asyncio.set_event_loop(loop)
|
124 |
+
result = loop.run_until_complete(translate(input_data))
|
125 |
+
loop.close()
|
126 |
|
127 |
# Display result
|
128 |
st.success("Translation:")
|