Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -4,15 +4,15 @@ import litellm
|
|
| 4 |
|
| 5 |
DESCRIPTION = '''
|
| 6 |
<div>
|
| 7 |
-
<h1 style="text-align: center;">TAIDE/
|
| 8 |
-
<p>This Space demonstrates the instruction-tuned model <a href="https://huggingface.co/taide/
|
| 9 |
</div>
|
| 10 |
'''
|
| 11 |
|
| 12 |
LICENSE = """
|
| 13 |
<p/>
|
| 14 |
---
|
| 15 |
-
Built with
|
| 16 |
"""
|
| 17 |
|
| 18 |
css = """
|
|
@@ -44,7 +44,7 @@ def chat(message: str,
|
|
| 44 |
messages.append({"role": "user", "content": message})
|
| 45 |
|
| 46 |
response = litellm.completion(
|
| 47 |
-
model="openai/
|
| 48 |
messages=messages,
|
| 49 |
max_completion_tokens=max_new_tokens,
|
| 50 |
temperature=temperature,
|
|
|
|
| 4 |
|
| 5 |
DESCRIPTION = '''
|
| 6 |
<div>
|
| 7 |
+
<h1 style="text-align: center;">TAIDE/TAIDE-LX-7B-Chat</h1>
|
| 8 |
+
<p>This Space demonstrates the instruction-tuned model <a href="https://huggingface.co/taide/TAIDE-LX-7B-Chat"><b>TAIDE-LX-7B-Chat</b></a>. TAIDE-LX-7B-Chat is the new open LLM and comes in one sizes: 8b. Feel free to play with it, or duplicate to run privately!</p>
|
| 9 |
</div>
|
| 10 |
'''
|
| 11 |
|
| 12 |
LICENSE = """
|
| 13 |
<p/>
|
| 14 |
---
|
| 15 |
+
Built with TAIDE-LX-7B-Chat
|
| 16 |
"""
|
| 17 |
|
| 18 |
css = """
|
|
|
|
| 44 |
messages.append({"role": "user", "content": message})
|
| 45 |
|
| 46 |
response = litellm.completion(
|
| 47 |
+
model="openai/TAIDE-LX-7B-Chat", # tells litellm to call the model via the Responses API
|
| 48 |
messages=messages,
|
| 49 |
max_completion_tokens=max_new_tokens,
|
| 50 |
temperature=temperature,
|