Spaces:

Jaaccaa
/

Data_augmentation

Sleeping

Jaaccaa commited on Sep 5

Commit

6602d3a

verified ·

1 Parent(s): 5cfea1d

Update backend/main.py

Files changed (1) hide show

backend/main.py CHANGED Viewed

@@ -101,7 +101,7 @@ async def generate_paraphrase(request: AugmentRequest):
             chat_completion = groq_client.chat.completions.create(
                 messages=[{"role": "user", "content": prompt}],
-                model="meta-llama/llama-prompt-guard-2-22m",
                 temperature=0.7,
                 max_tokens=100,
             )

             chat_completion = groq_client.chat.completions.create(
                 messages=[{"role": "user", "content": prompt}],
+                model="openai/gpt-oss-120b",
                 temperature=0.7,
                 max_tokens=100,
             )