Lhumpal commited on
Commit
f0ed595
·
verified ·
1 Parent(s): 6b67a64

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -5
app.py CHANGED
@@ -58,11 +58,8 @@ async def chat(request: ChatRequest):
58
  {"role": "user", "parts": [request.message]},
59
  ]
60
 
61
- response = client.models.generate_content(
62
- model="gemini-2.0-flash", contents=request.message
63
- )
64
- if response and hasattr(response, 'text'):
65
- return response.text # Return text if it's present
66
 
67
  except Exception as e:
68
  raise HTTPException(status_code=500, detail=str(e))
 
58
  {"role": "user", "parts": [request.message]},
59
  ]
60
 
61
+ response = client.models.generate_content(model="gemini-2.0-flash", contents=request.message)
62
+ return response.text
 
 
 
63
 
64
  except Exception as e:
65
  raise HTTPException(status_code=500, detail=str(e))