TaruniSwathi commited on
Commit
2af3ad8
·
verified ·
1 Parent(s): b9a2f08

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -141,14 +141,14 @@ def generate_code(task: str, user_input: str) -> str:
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}"
 
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}"