Hugging Face's logo

Spaces:
studzinsky
/
bielik_app_service
Sleeping

App Files Community
1
Patryk Studzinski commited on
Commit
a1c0774
·
1 Parent(s): 6348ce6

Fix: correct parameter name model_name_or_path

Browse files
Files changed (1) hide show
  1. app/main.py +1 -1
app/main.py CHANGED
@@ -33,7 +33,7 @@ app.add_middleware(
33
  # Global service initialization
34
  MODEL_PATH_IN_CONTAINER = "/app/pretrain_model"
35
  hf_service = HuggingFaceTextGenerationService(
36
- model_name_or_PATH=MODEL_PATH_IN_CONTAINER,
37
  device="cpu"
38
  )
39
 
 
33
  # Global service initialization
34
  MODEL_PATH_IN_CONTAINER = "/app/pretrain_model"
35
  hf_service = HuggingFaceTextGenerationService(
36
+ model_name_or_path=MODEL_PATH_IN_CONTAINER,
37
  device="cpu"
38
  )
39