acecalisto3 commited on
Commit
459114f
·
verified ·
1 Parent(s): 2b69de7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -6
app.py CHANGED
@@ -135,18 +135,15 @@ available_models = {
135
  "Code Generation & Completion": {
136
  "Salesforce CodeGen-350M (Mono)": pipeline("text-generation", model="Salesforce/codegen-350M-mono"),
137
  "BigCode StarCoder": pipeline("text-generation", model="bigcode/starcoder"),
138
- "Mixtral-8x7B-v0.1": pipeline("text-generation", model="mistralai/Mixtral-8x7B-v0.1"),
139
  "CodeGPT-small-py": pipeline("text-generation", model="microsoft/CodeGPT-small-py"),
140
  "PolyCoder-2.7B": pipeline("text-generation", model="NinedayWang/PolyCoder-2.7B"),
141
  "InCoder-1B": pipeline("text-generation", model="facebook/incoder-1B"),
142
- #... more code generation models
143
  },
144
  "Code Translation": {
145
- "Python to JavaScript": (lambda code, pipe=pipeline("translation", model="transformersbook/codeparrot-translation-en-java"): execute_translation(code, "javascript", pipe), []), # Pipeline for Python to JavaScript
146
- "Python to C++": (lambda code, pipe=pipeline("text-generation", model="konodyuk/codeparrot-small-trans-py-cpp"): execute_translation(code, "cpp", pipe), []), # Pipeline for Python to C++
147
- #... more language pairs
148
  },
149
- #... other categories
150
  }
151
 
152
  # Gradio interface with tabs
 
135
  "Code Generation & Completion": {
136
  "Salesforce CodeGen-350M (Mono)": pipeline("text-generation", model="Salesforce/codegen-350M-mono"),
137
  "BigCode StarCoder": pipeline("text-generation", model="bigcode/starcoder"),
 
138
  "CodeGPT-small-py": pipeline("text-generation", model="microsoft/CodeGPT-small-py"),
139
  "PolyCoder-2.7B": pipeline("text-generation", model="NinedayWang/PolyCoder-2.7B"),
140
  "InCoder-1B": pipeline("text-generation", model="facebook/incoder-1B"),
 
141
  },
142
  "Code Translation": {
143
+ "Python to JavaScript": (lambda code, pipe=pipeline("translation", model="transformersbook/codeparrot-translation-en-java"): execute_translation(code, "javascript", pipe), []),
144
+ "Python to C++": (lambda code, pipe=pipeline("text-generation", model="konodyuk/codeparrot-small-trans-py-cpp"): execute_translation(code, "cpp", pipe), []),
 
145
  },
146
+ # ... other categories
147
  }
148
 
149
  # Gradio interface with tabs