Aditya-1911 commited on
Commit
5545525
·
verified ·
1 Parent(s): 57398cd

Delete app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -23
app.py DELETED
@@ -1,23 +0,0 @@
1
- import os
2
- from huggingface_hub import InferenceClient
3
-
4
- # Initialize Hugging Face client
5
- client = InferenceClient(
6
- model="Salesforce/codegen-2B-multi",
7
- token=os.getenv("HUGGINGFACEHUB_API_TOKEN")
8
- )
9
-
10
- def generate_code(prompt):
11
- full_prompt = f"""You are a professional Python Data Scientist.
12
- Given the following instruction, write Python pandas code to perform the task on a dataframe named df.
13
- Instruction: {prompt}
14
- Output only valid Python code. No explanations."""
15
-
16
- response = client.text_generation(
17
- prompt=full_prompt,
18
- max_new_tokens=300,
19
- temperature=0.2,
20
- top_p=0.95,
21
- repetition_penalty=1.1
22
- )
23
- return response