Spaces:
Sleeping
Sleeping
arun3676 commited on
Commit ·
24f45dc
1
Parent(s): ff3ea6b
Add code analyzer endpoints with fine-tuned model
Browse files- hf-space/app.py +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 |
-
"
|
| 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
|