Guest User

Untitled

a guest
Jan 9th, 2024
89
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.00 KB | None | 0 0
  1. def generate_title(prompt: str) -> str:
  2. payload = {
  3. "model": "mistral-7b-instruct",
  4. "messages": [
  5. {
  6. "content": f"Generate title for a youtube video on the following topic:",
  7. "role": "system",
  8. "name": "system"
  9. },
  10. {
  11. "content": prompt,
  12. "role": "user",
  13. "name": "user"
  14. }
  15. ],
  16. "temperature": 0.7, # Adjust as needed
  17. "top_p": 1.0, # Adjust as needed
  18. "presence_penalty": 0.0, # Adjust as needed
  19. "frequency_penalty": "0.0" # Adjust as needed
  20. }
  21.  
  22. headers = {
  23. "Authorization": f"Bearer {llmos_api_key}",
  24. "Content-Type": "application/json"
  25. }
  26.  
  27. # Make the LLmos API request
  28. response = requests.post(url, json=payload, headers=headers)
  29. result = response.json()
  30.  
  31. # Extract the answer from the response
  32. video_title = result['choices'][0]['message']['content']
Advertisement
Add Comment
Please, Sign In to add comment