darshankr commited on
Commit
4bce69c
·
verified ·
1 Parent(s): 27cad93

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -2
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
- # Call translation function directly
121
- result = translate(input_data)
 
 
 
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:")