nikunjcepatel commited on
Commit
4394f5b
·
verified ·
1 Parent(s): 67fcd7c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -1
app.py CHANGED
@@ -23,7 +23,17 @@ def generate_text(input_text):
23
  "top_k": 0,
24
  })
25
  )
26
- return response
 
 
 
 
 
 
 
 
 
 
27
 
28
  # Create Gradio interface
29
  iface = gr.Interface(fn=generate_text, inputs="text", outputs="text")
 
23
  "top_k": 0,
24
  })
25
  )
26
+
27
+ # Handle errors
28
+ if response.status_code != 200:
29
+ return f"Error: {response.status_code}, {response.text}"
30
+
31
+ # Parse and return the content of the response
32
+ try:
33
+ response_json = response.json()
34
+ return response_json.get("choices", [{}])[0].get("message", {}).get("content", "No content returned.")
35
+ except json.JSONDecodeError:
36
+ return "Error: Unable to parse response."
37
 
38
  # Create Gradio interface
39
  iface = gr.Interface(fn=generate_text, inputs="text", outputs="text")