arun3676 commited on
Commit
24f45dc
·
1 Parent(s): ff3ea6b

Add code analyzer endpoints with fine-tuned model

Browse files
Files changed (2) hide show
  1. hf-space/app.py +2 -2
  2. hf-space/requirements.txt +2 -2
hf-space/app.py CHANGED
@@ -31,7 +31,7 @@ async def load_model():
31
 
32
  model = PeftModel.from_pretrained(
33
  base_model,
34
- "arun3676/fine-tuned-code-analyzer"
35
  )
36
  print("✅ Model loaded successfully!")
37
 
@@ -88,4 +88,4 @@ async def root():
88
  }
89
 
90
  if __name__ == "__main__":
91
- uvicorn.run(app, host="0.0.0.0", port=7860)
 
31
 
32
  model = PeftModel.from_pretrained(
33
  base_model,
34
+ "arunn7/fine-tuned-code-analyzer"
35
  )
36
  print("✅ Model loaded successfully!")
37
 
 
88
  }
89
 
90
  if __name__ == "__main__":
91
+ uvicorn.run(app, host="0.0.0.0", port=7860)
hf-space/requirements.txt CHANGED
@@ -1,7 +1,7 @@
1
  fastapi
2
- uvicorn
3
  transformers
4
  peft
5
  torch
6
  accelerate
7
- requests
 
1
  fastapi
2
+ uvicorn[standard]
3
  transformers
4
  peft
5
  torch
6
  accelerate
7
+ requests