Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -14,7 +14,7 @@ os.environ["TOKENIZERS_PARALLELISM"] = "false"
|
|
14 |
|
15 |
# Default settings
|
16 |
class ChatConfig:
|
17 |
-
MODEL = "google/gemma-
|
18 |
DEFAULT_SYSTEM_MSG = "You are an AI assistant answering only based on the uploaded PDF."
|
19 |
DEFAULT_MAX_TOKENS = 512
|
20 |
DEFAULT_TEMP = 0.3
|
@@ -98,7 +98,7 @@ def generate_response(
|
|
98 |
try:
|
99 |
# Use OpenRouter to get the response
|
100 |
completion = client.chat.completions.create(
|
101 |
-
model="google/gemma-
|
102 |
messages=messages
|
103 |
)
|
104 |
return completion.choices[0].message.content
|
|
|
14 |
|
15 |
# Default settings
|
16 |
class ChatConfig:
|
17 |
+
MODEL = "google/gemma-3-27b-it:free" # Use OpenRouter's Gemma model
|
18 |
DEFAULT_SYSTEM_MSG = "You are an AI assistant answering only based on the uploaded PDF."
|
19 |
DEFAULT_MAX_TOKENS = 512
|
20 |
DEFAULT_TEMP = 0.3
|
|
|
98 |
try:
|
99 |
# Use OpenRouter to get the response
|
100 |
completion = client.chat.completions.create(
|
101 |
+
model="google/gemma-3-27b-it:free",
|
102 |
messages=messages
|
103 |
)
|
104 |
return completion.choices[0].message.content
|