Not-Grim-Refer commited on
Commit
e0903e9
1 Parent(s): a36c7d4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -4
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
- prompt += f"Logic: {' '.join(code_analysis['logic'])}"
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