Karan0310 commited on
Commit
e84dca6
·
verified ·
1 Parent(s): 7a7b63e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -8,7 +8,7 @@ import pycountry
8
  from datetime import datetime
9
  from pydantic import BaseModel, PydanticUserError, ConfigDict
10
  from pydantic import BaseModel, ConfigDict
11
-
12
 
13
  class MyModel(BaseModel):
14
  request: 'starlette.requests.Request'
@@ -23,7 +23,8 @@ Request.__get_pydantic_core_schema__ = get_pydantic_core_schema
23
  # Initialize pipelines with error handling
24
  try:
25
  lang_detector = pipeline("text-classification", model="papluca/xlm-roberta-base-language-detection")
26
- text_translator = pipeline("translation", model="facebook/nllb-200-distilled-600M", torch_dtype=torch.bfloat16)
 
27
  print("🚀 AI Translation Hub initialized successfully!")
28
  except Exception as e:
29
  print(f"⚠️ Error initializing models: {e}")
 
8
  from datetime import datetime
9
  from pydantic import BaseModel, PydanticUserError, ConfigDict
10
  from pydantic import BaseModel, ConfigDict
11
+ import json
12
 
13
  class MyModel(BaseModel):
14
  request: 'starlette.requests.Request'
 
23
  # Initialize pipelines with error handling
24
  try:
25
  lang_detector = pipeline("text-classification", model="papluca/xlm-roberta-base-language-detection")
26
+ #text_translator = pipeline("translation", model="facebook/nllb-200-distilled-600M", torch_dtype=torch.bfloat16)
27
+ text_translator = pipeline("translation", model="facebook/nllb-200-distilled-600M")
28
  print("🚀 AI Translation Hub initialized successfully!")
29
  except Exception as e:
30
  print(f"⚠️ Error initializing models: {e}")