Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -20,7 +20,7 @@ def load_model():
|
|
| 20 |
|
| 21 |
model = pipeline(
|
| 22 |
"text-generation",
|
| 23 |
-
model="kacperbb/phi-3.5-
|
| 24 |
trust_remote_code=True
|
| 25 |
)
|
| 26 |
logger.info("✅ YOUR fine-tuned model loaded successfully!")
|
|
@@ -79,7 +79,7 @@ def health():
|
|
| 79 |
def home():
|
| 80 |
return jsonify({
|
| 81 |
"message": "Phi 3.5 Fine-tuned API is running!",
|
| 82 |
-
"model": "kacperbb/phi-3.5-
|
| 83 |
"endpoints": {
|
| 84 |
"generate": "POST /generate",
|
| 85 |
"health": "GET /health"
|
|
|
|
| 20 |
|
| 21 |
model = pipeline(
|
| 22 |
"text-generation",
|
| 23 |
+
model="kacperbb/phi-3.5-merged-lora",
|
| 24 |
trust_remote_code=True
|
| 25 |
)
|
| 26 |
logger.info("✅ YOUR fine-tuned model loaded successfully!")
|
|
|
|
| 79 |
def home():
|
| 80 |
return jsonify({
|
| 81 |
"message": "Phi 3.5 Fine-tuned API is running!",
|
| 82 |
+
"model": "kacperbb/phi-3.5-merged-lora",
|
| 83 |
"endpoints": {
|
| 84 |
"generate": "POST /generate",
|
| 85 |
"health": "GET /health"
|