Spaces:
Runtime error
Runtime error
Not-Grim-Refer
commited on
Commit
•
e0903e9
1
Parent(s):
a36c7d4
Update app.py
Browse files
app.py
CHANGED
@@ -1,6 +1,3 @@
|
|
1 |
-
import os
|
2 |
-
os.system("pip install -r requirements.txt")
|
3 |
-
os.system("pip freeze")
|
4 |
import torch
|
5 |
from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
|
6 |
import gradio as gr
|
@@ -34,7 +31,7 @@ def generate_prompt(code_analysis):
|
|
34 |
prompt = f"Generate code with the following: \n\n"
|
35 |
prompt += f"Variables: {', '.join(code_analysis['variables'])} \n\n"
|
36 |
prompt += f"Functions: {', '.join(code_analysis['functions'])} \n\n"
|
37 |
-
|
38 |
return prompt
|
39 |
|
40 |
# Generate code from model and prompt
|
|
|
|
|
|
|
|
|
1 |
import torch
|
2 |
from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
|
3 |
import gradio as gr
|
|
|
31 |
prompt = f"Generate code with the following: \n\n"
|
32 |
prompt += f"Variables: {', '.join(code_analysis['variables'])} \n\n"
|
33 |
prompt += f"Functions: {', '.join(code_analysis['functions'])} \n\n"
|
34 |
+
prompt += f"Logic: {' '.join(code_analysis['logic'])}"
|
35 |
return prompt
|
36 |
|
37 |
# Generate code from model and prompt
|