mjolnir1122 commited on
Commit
d6bd667
·
verified ·
1 Parent(s): e07cf59

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -61,7 +61,7 @@ def rag_pipeline(query):
61
  model="llama3-70b-8192"
62
  )
63
 
64
- return response['choices'][0]['message']['content']
65
 
66
  # Streamlit UI
67
  st.set_page_config(page_title="HydroGen-AI", layout="wide")
 
61
  model="llama3-70b-8192"
62
  )
63
 
64
+ return response.choices[0].message.content # ✅ Fix applied
65
 
66
  # Streamlit UI
67
  st.set_page_config(page_title="HydroGen-AI", layout="wide")