Update app.py
Browse files
app.py
CHANGED
|
@@ -1,6 +1,5 @@
|
|
| 1 |
import gradio as gr
|
| 2 |
import requests
|
| 3 |
-
import os
|
| 4 |
|
| 5 |
# Hardcoded Groq API key (for testing only, remove later)
|
| 6 |
GROQ_API_KEY = "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"
|
|
@@ -26,10 +25,10 @@ def chat_with_groq(message, history):
|
|
| 26 |
|
| 27 |
# Make sure the URL and model are correct
|
| 28 |
print(f"API URL: {API_URL}")
|
| 29 |
-
print(f"Model being used:
|
| 30 |
|
| 31 |
payload = {
|
| 32 |
-
"model": "
|
| 33 |
"messages": [{"role": "system", "content": SYSTEM_PROMPT}] + [
|
| 34 |
{"role": "user", "content": msg[0]} for msg in history
|
| 35 |
] + [{"role": "user", "content": message}],
|
|
@@ -68,3 +67,4 @@ with gr.Blocks() as demo:
|
|
| 68 |
|
| 69 |
demo.launch()
|
| 70 |
|
|
|
|
|
|
| 1 |
import gradio as gr
|
| 2 |
import requests
|
|
|
|
| 3 |
|
| 4 |
# Hardcoded Groq API key (for testing only, remove later)
|
| 5 |
GROQ_API_KEY = "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"
|
|
|
|
| 25 |
|
| 26 |
# Make sure the URL and model are correct
|
| 27 |
print(f"API URL: {API_URL}")
|
| 28 |
+
print(f"Model being used: gemma-7b-it") # Changed model to gemma-7b-it
|
| 29 |
|
| 30 |
payload = {
|
| 31 |
+
"model": "gemma-7b-it", # Changed model here
|
| 32 |
"messages": [{"role": "system", "content": SYSTEM_PROMPT}] + [
|
| 33 |
{"role": "user", "content": msg[0]} for msg in history
|
| 34 |
] + [{"role": "user", "content": message}],
|
|
|
|
| 67 |
|
| 68 |
demo.launch()
|
| 69 |
|
| 70 |
+
|