tbaig1605 commited on
Commit
4ed6fcb
·
verified ·
1 Parent(s): edafe5a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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: mixtral-8x7b-32768")
30
 
31
  payload = {
32
- "model": "mixtral-8x7b-32768", # Ensure using a valid 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
+