MiCkSoftware commited on
Commit
e2cdbc6
·
1 Parent(s): 0d3bbff

testing gpt neo

Browse files
Files changed (2) hide show
  1. app.py +3 -1
  2. client.py +3 -3
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'ypothenuse pour un triangle de cote 4 et 9",
9
  "history": [],
10
- "system_message": "You are a friendly Chatbot.",
11
  "max_tokens": 512,
12
- "temperature": 0.7,
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