Spaces:
Paused
Paused
Commit ·
e2cdbc6
1
Parent(s): 0d3bbff
testing gpt neo
Browse files
app.py
CHANGED
|
@@ -6,7 +6,9 @@ from typing import List, Tuple
|
|
| 6 |
import asyncio
|
| 7 |
|
| 8 |
# Initialisation du client Hugging Face
|
| 9 |
-
client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
|
|
|
|
|
|
|
| 10 |
|
| 11 |
# Initialisation de FastAPI
|
| 12 |
app = FastAPI()
|
|
|
|
| 6 |
import asyncio
|
| 7 |
|
| 8 |
# Initialisation du client Hugging Face
|
| 9 |
+
# client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
|
| 10 |
+
client = InferenceClient("EleutherAI/gpt-neo-1.3B")
|
| 11 |
+
|
| 12 |
|
| 13 |
# Initialisation de FastAPI
|
| 14 |
app = FastAPI()
|
client.py
CHANGED
|
@@ -5,11 +5,11 @@ url = "https://micksoftware-laria-startup.hf.space/predict"
|
|
| 5 |
|
| 6 |
# Données pour la requête
|
| 7 |
payload = {
|
| 8 |
-
"message": "quelle longueur d'
|
| 9 |
"history": [],
|
| 10 |
-
"system_message": "You are a friendly
|
| 11 |
"max_tokens": 512,
|
| 12 |
-
"temperature": 0.
|
| 13 |
"top_p": 0.95,
|
| 14 |
}
|
| 15 |
|
|
|
|
| 5 |
|
| 6 |
# Données pour la requête
|
| 7 |
payload = {
|
| 8 |
+
"message": "quelle longueur d'hypothenuse pour un triangle de cote 4 et 9",
|
| 9 |
"history": [],
|
| 10 |
+
"system_message": "You are a friendly assistant. Answer questions briefly and do not continue conversations unless explicitly asked.",
|
| 11 |
"max_tokens": 512,
|
| 12 |
+
"temperature": 0.4,
|
| 13 |
"top_p": 0.95,
|
| 14 |
}
|
| 15 |
|