nurqoneah commited on
Commit
e708fb6
·
verified ·
1 Parent(s): ea68007

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -65,7 +65,7 @@ def create_llm():
65
  # return HuggingFacePipeline(pipeline=text_generation_pipeline)
66
  return HuggingFaceHub(
67
  repo_id=MODEL_NAME,
68
- task="text-generation"
69
  model_kwargs={
70
  "temperature": 0.7, # Balanced between creativity and accuracy
71
  "max_new_tokens": 1024,
 
65
  # return HuggingFacePipeline(pipeline=text_generation_pipeline)
66
  return HuggingFaceHub(
67
  repo_id=MODEL_NAME,
68
+ task="text-generation",
69
  model_kwargs={
70
  "temperature": 0.7, # Balanced between creativity and accuracy
71
  "max_new_tokens": 1024,