Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -141,14 +141,14 @@ def generate_code(task: str, user_input: str) -> str:
|
|
| 141 |
skip_special_tokens=True,
|
| 142 |
)
|
| 143 |
|
| 144 |
-
|
| 145 |
|
| 146 |
-
if not
|
| 147 |
return "The model returned an empty response. Please try again."
|
| 148 |
|
| 149 |
language = "java" if task == "Natural Language → Java" else "csharp"
|
| 150 |
|
| 151 |
-
return f"```{language}\n{
|
| 152 |
|
| 153 |
except Exception as error:
|
| 154 |
return f"Generation failed: {type(error).__name__}: {error}"
|
|
|
|
| 141 |
skip_special_tokens=True,
|
| 142 |
)
|
| 143 |
|
| 144 |
+
code_ = _clean_csharp_output(generated_text)
|
| 145 |
|
| 146 |
+
if not code_:
|
| 147 |
return "The model returned an empty response. Please try again."
|
| 148 |
|
| 149 |
language = "java" if task == "Natural Language → Java" else "csharp"
|
| 150 |
|
| 151 |
+
return f"```{language}\n{code_}\n```"
|
| 152 |
|
| 153 |
except Exception as error:
|
| 154 |
return f"Generation failed: {type(error).__name__}: {error}"
|