Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -65,11 +65,14 @@ def call_ollama_api(model, prompt):
|
|
| 65 |
"content": prompt
|
| 66 |
}
|
| 67 |
],
|
| 68 |
-
timeout=
|
| 69 |
)
|
| 70 |
return response.choices[0].message.content
|
|
|
|
|
|
|
|
|
|
| 71 |
except Exception as e:
|
| 72 |
-
logger.error(f"Error calling Ollama API for model {model}: {e}")
|
| 73 |
return f"Error: Unable to get response from the model."
|
| 74 |
|
| 75 |
# Generate responses using two randomly selected models
|
|
|
|
| 65 |
"content": prompt
|
| 66 |
}
|
| 67 |
],
|
| 68 |
+
timeout=180
|
| 69 |
)
|
| 70 |
return response.choices[0].message.content
|
| 71 |
+
except requests.exceptions.Timeout:
|
| 72 |
+
logger.error(f"Timeout error for model {model} after 180 seconds")
|
| 73 |
+
return f"Error: Model response timed out after 180 seconds"
|
| 74 |
except Exception as e:
|
| 75 |
+
logger.error(f"Error calling Ollama API for model {model}: {str(e)}")
|
| 76 |
return f"Error: Unable to get response from the model."
|
| 77 |
|
| 78 |
# Generate responses using two randomly selected models
|