Spaces:
Runtime error
Runtime error
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
|