kacperbb commited on
Commit
2acc762
·
verified ·
1 Parent(s): 4245fe0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -20,7 +20,7 @@ def load_model():
20
 
21
  model = pipeline(
22
  "text-generation",
23
- model="kacperbb/phi-3.5-hf-finetuned",
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-hf-finetuned",
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"