Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- def generate_title(prompt: str) -> str:
- payload = {
- "model": "mistral-7b-instruct",
- "messages": [
- {
- "content": f"Generate title for a youtube video on the following topic:",
- "role": "system",
- "name": "system"
- },
- {
- "content": prompt,
- "role": "user",
- "name": "user"
- }
- ],
- "temperature": 0.7, # Adjust as needed
- "top_p": 1.0, # Adjust as needed
- "presence_penalty": 0.0, # Adjust as needed
- "frequency_penalty": "0.0" # Adjust as needed
- }
- headers = {
- "Authorization": f"Bearer {llmos_api_key}",
- "Content-Type": "application/json"
- }
- # Make the LLmos API request
- response = requests.post(url, json=payload, headers=headers)
- result = response.json()
- # Extract the answer from the response
- video_title = result['choices'][0]['message']['content']
Advertisement
Add Comment
Please, Sign In to add comment