File size: 801 Bytes
7898606
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
import requests

API_URL = "https://api-inference.huggingface.co/models/mistralai/Mistral-7B-Instruct-v0.2"
headers = {"Authorization": f"Bearer your_token_here"}  # Replace with your actual token or use os.getenv

def explain_code(code):
    prompt = f"Explain this Python code in simple terms:\n{code}"
    res = requests.post(API_URL, headers=headers, json={"inputs": prompt})
    return res.json()[0]["generated_text"]

def debug_code(code):
    prompt = f"Find bugs in this code:\n{code}"
    res = requests.post(API_URL, headers=headers, json={"inputs": prompt})
    return res.json()[0]["generated_text"]

def ask_question(q):
    prompt = f"Answer this Python question:\n{q}"
    res = requests.post(API_URL, headers=headers, json={"inputs": prompt})
    return res.json()[0]["generated_text"]