Upshivam commited on
Commit
5e0018c
·
1 Parent(s): 78e641c

Fix: switch to Llama-3.2-3B-Instruct chat model

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -3,7 +3,7 @@ import os
3
  from huggingface_hub import InferenceClient
4
 
5
  client = InferenceClient(
6
- model="mistralai/Mistral-7B-Instruct-v0.3",
7
  token=os.environ.get("HF_TOKEN")
8
  )
9
 
@@ -11,7 +11,7 @@ def generate_dockerfile(description, base_os, app_type):
11
  messages = [
12
  {
13
  "role": "system",
14
- "content": "You are a DevOps expert. Return ONLY Dockerfile content, no explanation."
15
  },
16
  {
17
  "role": "user",
 
3
  from huggingface_hub import InferenceClient
4
 
5
  client = InferenceClient(
6
+ model="meta-llama/Llama-3.2-3B-Instruct",
7
  token=os.environ.get("HF_TOKEN")
8
  )
9
 
 
11
  messages = [
12
  {
13
  "role": "system",
14
+ "content": "You are a DevOps expert. Return ONLY Dockerfile content, no explanation, no markdown."
15
  },
16
  {
17
  "role": "user",